| .. |
|
parallel_end_to_end
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
__init__.py
|
implementing-searching-strategy-for-inference
|
2020-05-07 14:53:50 +08:00 |
|
test_add_relu_redistribution.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_allreduce_fusion.py
|
enable/disable allreduce_fusion
|
2020-04-24 10:34:59 +08:00 |
|
test_alltoall.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_arithmetic.py
|
fix refkey bug for auto parallel
|
2020-04-02 16:14:29 +08:00 |
|
test_auto_parallel_BN_PReLU.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_arithmetic.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_assign_sub_with_ref_key.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_cast.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_common_parameter.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_double_star.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_double_subgraphs.py
|
support-the-multiple-subgraphs-in-the-ANF
|
2020-04-27 09:34:41 +08:00 |
|
test_auto_parallel_fc_nobias.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_four_matmul.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_inference.py
|
implementing-searching-strategy-for-inference
|
2020-05-07 14:53:50 +08:00 |
|
test_auto_parallel_l2normalize.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_matmul_drop.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_matmul_prelu.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_onehot.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_parameter_cast.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_partial_strategy.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_reduce_method.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_reshape.py
|
modify longtime python ut
|
2020-03-31 16:18:04 +08:00 |
|
test_auto_parallel_resnet.py
|
remove enbale hccl
|
2020-04-28 10:37:08 +08:00 |
|
test_auto_parallel_rhombus.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_softmax_loss.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_transformer.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_transpose.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_auto_parallel_tuple_depend.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_two_bn.py
|
support-the-multiple-subgraphs-in-the-ANF
|
2020-04-27 09:34:41 +08:00 |
|
test_auto_parallel_two_matmul.py
|
delete the 'simplify_cal' attribute in 'set_algo_parameters' and 'get_algo_parameters' interface
|
2020-04-24 10:46:38 +08:00 |
|
test_auto_parallel_two_partial_matmul.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_parallel_zig_zag.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_auto_star_elimination.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_batch_matmul.py
|
add parallel ops for neg and batchmatmul
|
2020-04-01 09:36:38 +08:00 |
|
test_batch_parallel.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_batch_parallel_dropout.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_batch_parallel_tensoradd.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_batchnorm_batch_parallel.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_bias_add.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_bn_prelu_cell.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_bool_grad.py
|
fix bug in infer_dtype function of hcom operations
|
2020-04-27 18:49:21 +08:00 |
|
test_broadcast_dict.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_combined_tensor.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_comparison_function_info.py
|
add minimum distributed op
|
2020-04-13 17:26:31 +08:00 |
|
test_dataset.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_dataset_interface.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_dataset_util.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_dense_matmul.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_different_type_for_div_op.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_dropout_do_mask.py
|
add parallel op for dropoutdomask
|
2020-04-11 12:11:22 +08:00 |
|
test_element_wise_function.py
|
Support pow's second input could be tensor and fix bug in bprop of pow
|
2020-04-11 16:16:58 +08:00 |
|
test_expand_dims.py
|
add parallel ops for expand dims
|
2020-04-03 15:17:07 +08:00 |
|
test_forward_graph.py
|
support forward graph
|
2020-04-26 14:44:23 +08:00 |
|
test_gather_v2_primitive.py
|
support distributed GatherV2 operator
|
2020-04-12 16:00:47 +08:00 |
|
test_get_next.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_get_parameter_layout.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_hybird_parallel_activation.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_l2normalize.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_layer_norm.py
|
fix layernorm bug
|
2020-04-26 16:39:32 +08:00 |
|
test_linear.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_load_tensor.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_loop_two_matmul.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_loss_and_optimizer.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_matmul_dropout.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_matmul_tensor.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_mix_precision_hybrid_parallel.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_model_callback.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_neg.py
|
add parallel ops for neg and batchmatmul
|
2020-04-01 09:36:38 +08:00 |
|
test_one_dev.py
|
remove ge depend in cpu
|
2020-04-02 19:38:42 +08:00 |
|
test_one_hot_net.py
|
modify longtime python ut
|
2020-03-31 16:18:04 +08:00 |
|
test_one_weight_parameter.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_onehot.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_operator_model_parallel.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_optimizer.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_optimizer_clone_weight.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_parameter_init.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_prelu.py
|
auto parallel prelu support prelu
|
2020-04-24 14:10:57 +08:00 |
|
test_prelu_cell.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_reduce_method_info.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_reshape.py
|
Change 'NOT_FULLY_USE_DEVICES' to 'FULLY_USE_DEVICES' and make ALL-1 user-specified-strategy valid in auto-parallel
|
2020-04-15 10:14:46 +08:00 |
|
test_reshape_parameter.py
|
support reshape parameter
|
2020-04-24 15:59:09 +08:00 |
|
test_scalar_loss.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_semi_auto_two_subgraphs.py
|
support multiple subgraphs
|
2020-04-22 17:40:56 +08:00 |
|
test_set_auto_parallel_context.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_sigmoid_cross_entropy_with_logits.py
|
add parallel op for sigmoidloss
|
2020-04-16 14:33:24 +08:00 |
|
test_softmax_cross_entropy_expand.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_softmax_cross_entropy_loss.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_split_grad_sens.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_square.py
|
add parallel op for square
|
2020-04-17 10:42:52 +08:00 |
|
test_squeeze_info.py
|
add squeeze distributed op
|
2020-04-11 16:03:08 +08:00 |
|
test_step_parallel.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_strategy_checkpoint.py
|
use param name as the key of strategy checkpoint
|
2020-04-30 11:01:32 +08:00 |
|
test_sum_as_loss.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_transpose.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_two_matmul.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_two_weights_parameter.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_using_seed_for_initializer.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
test_virtual_dataset_3_input.py
|
initial version
|
2020-03-27 22:54:54 +08:00 |
Xiaoda Zhang
def8573275