mindspore2022/mindspore/ccsrc/parallel/ops_info
Xiaoda Zhang 0ac50a19f5 Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
..
activation_info.cc add squeeze distributed op 2020-04-11 16:03:08 +08:00
activation_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
arithmetic_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
arithmetic_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
batch_parallel_info.cc support distributed GatherV2 operator 2020-04-12 16:00:47 +08:00
batch_parallel_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
bias_add_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
bias_add_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
comparison_function_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
dropout_do_mask_info.cc add parallel op for dropoutdomask 2020-04-11 12:11:22 +08:00
dropout_do_mask_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
elementary_function_info.h Support pow's second input could be tensor and fix bug in bprop of pow 2020-04-11 16:16:58 +08:00
gather_v2_info.cc support distributed GatherV2 operator 2020-04-12 16:00:47 +08:00
gather_v2_info.h support distributed GatherV2 operator 2020-04-12 16:00:47 +08:00
get_next_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
get_next_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
l2_normalize_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
l2_normalize_info.h refactor OperatorCostPtr in OperatorInfo 2020-04-09 20:37:52 +08:00
loss_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
loss_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
matmul_info.cc Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
matmul_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
onehot_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
onehot_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
operator_info.cc Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
operator_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
ops_info_head_files.h support distributed GatherV2 operator 2020-04-12 16:00:47 +08:00
ops_utils.h !245 Add bool type check in communication operator 2020-04-13 20:19:47 +08:00
prelu_info.cc fix auto parallel prelu 2020-04-08 20:45:08 +08:00
prelu_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
reduce_method_info.cc Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
reduce_method_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
reshape_info.cc initial version 2020-03-27 22:54:54 +08:00
reshape_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
tmp_identity_info.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
tmp_identity_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
transpose_info.cc initial version 2020-03-27 22:54:54 +08:00
transpose_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00
virtual_dataset_info.cc initial version 2020-03-27 22:54:54 +08:00
virtual_dataset_info.h Model the memory cost in auto-parallel. It is calculated by the output of operators, plus the parameters. Additionally, modify the graph-operations in auto_parallel to include memory_cost. 2020-04-14 11:39:31 +08:00