mindspore2022/mindspore/ccsrc/parallel/tensor_layout
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
..
arrangement.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
arrangement.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
array.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
array.h initial version 2020-03-27 22:54:54 +08:00
construct_operator.cc initial version 2020-03-27 22:54:54 +08:00
construct_operator.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
layout_transfer.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
layout_transfer.h initial version 2020-03-27 22:54:54 +08:00
map.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
map.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
redistribution_layout_transfer.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
redistribution_layout_transfer.h initial version 2020-03-27 22:54:54 +08:00
redistribution_operator_infer.cc initial version 2020-03-27 22:54:54 +08:00
redistribution_operator_infer.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
reshape_layout_transfer.cc initial version 2020-03-27 22:54:54 +08:00
reshape_layout_transfer.h initial version 2020-03-27 22:54:54 +08:00
shape_util.cc initial version 2020-03-27 22:54:54 +08:00
shape_util.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
tensor_info.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
tensor_layout.cc sort include file in parallel dir 2020-04-02 15:14:47 +08:00
tensor_layout.h sort include file in parallel dir 2020-04-02 15:14:47 +08:00
tensor_redistribution.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
tensor_redistribution.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