| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.h | 71 …virtual double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &o…
73 …virtual double GetForwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI…
75 …irtual double GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI…
78 …virtual double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI…
80 virtual double GetForwardComputationCost(const std::vector<TensorInfo> &inputs,
81 … const std::vector<TensorInfo> &outputs, int64_t stage_id) const = 0;
82 virtual double GetBackwardComputationCost(const std::vector<TensorInfo> &inputs,
83 … const std::vector<TensorInfo> &outputs, int64_t stage_id) const = 0;
90 …virtual double GetMemoryCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> …
92 …double GetInputMemoryCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou…
[all …]
|
| D | operator_costmodel.cc | 43 double OperatorCost::GetMemoryCost(const std::vector<TensorInfo> &inputs, in GetMemoryCost()
44 const std::vector<TensorInfo> &outputs) const { in GetMemoryCost()
48 … OperatorCost::GetInputMemoryCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI… in GetInputMemoryCost()
58 double OperatorCost::GetOutputMemoryCost(const std::vector<TensorInfo> &, in GetOutputMemoryCost() argument
59 const std::vector<TensorInfo> &outputs) const { in GetOutputMemoryCost()
70 double OperatorCost::GetMemoryCostForInference(const std::vector<TensorInfo> &, in GetMemoryCostForInference() argument
71 const std::vector<TensorInfo> &outputs) const { in GetMemoryCostForInference()
85 …le MatMulCost::GetForwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI… in GetForwardCommCost()
87 TensorInfo input0 = inputs[0]; in GetForwardCommCost()
88 TensorInfo output0 = outputs[0]; in GetForwardCommCost()
[all …]
|
| D | edge_costmodel.h | 94 …put(const std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> &output_set) { in set_pre_op_output()
97 …nput(const std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> &input_set) { in set_next_op_input()
104 …std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> prev_op_output() const { in prev_op_output()
107 std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> next_op_input() const { in next_op_input()
154 std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> pre_op_output_;
155 std::vector<std::pair<std::shared_ptr<Strategy>, std::vector<TensorInfo>>> next_op_input_;
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | onehot_info_test_axis_0.cc | 108 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
109 std::vector<TensorInfo> outputs = onehot_info2->outputs_tensor_info(); in TEST_F()
114 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
115 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
130 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
131 std::vector<TensorInfo> outputs = onehot_info2->outputs_tensor_info(); in TEST_F()
136 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
137 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
152 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
153 std::vector<TensorInfo> outputs = onehot_info2->outputs_tensor_info(); in TEST_F()
[all …]
|
| D | matmul_info_test.cc | 171 std::vector<TensorInfo> inputs = matmul1->inputs_tensor_info(); in TEST_F()
172 std::vector<TensorInfo> outputs = matmul1->outputs_tensor_info(); in TEST_F()
178 TensorInfo mat_a_tensor_info = inputs.at(0); in TEST_F()
179 TensorInfo mat_b_tensor_info = inputs.at(1); in TEST_F()
180 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
197 std::vector<TensorInfo> inputs = matmul2->inputs_tensor_info(); in TEST_F()
198 std::vector<TensorInfo> outputs = matmul2->outputs_tensor_info(); in TEST_F()
204 TensorInfo mat_a_tensor_info = inputs.at(0); in TEST_F()
205 TensorInfo mat_b_tensor_info = inputs.at(1); in TEST_F()
206 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
[all …]
|
| D | reshape_test.cc | 97 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
98 std::vector<TensorInfo> outputs = reshape->outputs_tensor_info(); in TEST_F()
103 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
104 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
118 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
119 std::vector<TensorInfo> outputs = reshape->outputs_tensor_info(); in TEST_F()
124 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
125 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
139 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
140 std::vector<TensorInfo> outputs = reshape->outputs_tensor_info(); in TEST_F()
[all …]
|
| D | onehot_info_test.cc | 108 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
109 std::vector<TensorInfo> outputs = onehot_info->outputs_tensor_info(); in TEST_F()
114 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
115 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
130 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
131 std::vector<TensorInfo> outputs = onehot_info->outputs_tensor_info(); in TEST_F()
136 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
137 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
152 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
153 std::vector<TensorInfo> outputs = onehot_info->outputs_tensor_info(); in TEST_F()
[all …]
|
| D | prelu_test.cc | 82 std::vector<TensorInfo> inputs = prelu->inputs_tensor_info(); in TEST_F()
83 std::vector<TensorInfo> outputs = prelu->outputs_tensor_info(); in TEST_F()
89 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
90 TensorInfo param_tensor_info = inputs.at(1); in TEST_F()
91 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
105 std::vector<TensorInfo> inputs = prelu->inputs_tensor_info(); in TEST_F()
106 std::vector<TensorInfo> outputs = prelu->outputs_tensor_info(); in TEST_F()
112 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
113 TensorInfo param_tensor_info = inputs.at(1); in TEST_F()
114 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
[all …]
|
| D | softmax_entropy_loss_info_test.cc | 82 std::vector<TensorInfo> inputs = loss->inputs_tensor_info(); in TEST_F()
83 std::vector<TensorInfo> outputs = loss->outputs_tensor_info(); in TEST_F()
90 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
91 TensorInfo label_tensor_info = inputs.at(1); in TEST_F()
92 TensorInfo output_0_tensor_info = outputs.at(0); in TEST_F()
93 TensorInfo output_1_tensor_info = outputs.at(1); in TEST_F()
111 std::vector<TensorInfo> inputs = loss->inputs_tensor_info(); in TEST_F()
112 std::vector<TensorInfo> outputs = loss->outputs_tensor_info(); in TEST_F()
119 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
120 TensorInfo label_tensor_info = inputs.at(1); in TEST_F()
[all …]
|
| D | tensor_add_info_test.cc | 84 std::vector<TensorInfo> inputs = tensor_add->inputs_tensor_info(); in TEST_F()
85 std::vector<TensorInfo> outputs = tensor_add->outputs_tensor_info(); in TEST_F()
90 TensorInfo inputa_tensor_info = inputs.at(0); in TEST_F()
91 TensorInfo inputb_tensor_info = inputs.at(1); in TEST_F()
92 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
108 std::vector<TensorInfo> inputs = tensor_add->inputs_tensor_info(); in TEST_F()
109 std::vector<TensorInfo> outputs = tensor_add->outputs_tensor_info(); in TEST_F()
114 TensorInfo inputa_tensor_info = inputs.at(0); in TEST_F()
115 TensorInfo inputb_tensor_info = inputs.at(1); in TEST_F()
116 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
[all …]
|
| D | tmpidentity_test.cc | 83 std::vector<TensorInfo> inputs = identity_ptr->inputs_tensor_info(); in TEST_F()
84 std::vector<TensorInfo> outputs = identity_ptr->outputs_tensor_info(); in TEST_F()
89 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
90 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
104 std::vector<TensorInfo> inputs = identity_ptr->inputs_tensor_info(); in TEST_F()
105 std::vector<TensorInfo> outputs = identity_ptr->outputs_tensor_info(); in TEST_F()
110 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
111 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
146 std::vector<TensorInfo> inputs_info = identity_ptr->inputs_tensor_info(); in TEST_F()
147 std::vector<TensorInfo> outputs_info = identity_ptr->outputs_tensor_info(); in TEST_F()
|
| D | tanh_info_test.cc | 81 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F()
82 std::vector<TensorInfo> outputs = tanh->outputs_tensor_info(); in TEST_F()
87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
88 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
102 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> outputs = tanh->outputs_tensor_info(); in TEST_F()
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
109 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | log_softmax_info_test.cc | 82 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F()
83 std::vector<TensorInfo> outputs = log_softmax->outputs_tensor_info(); in TEST_F()
88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
89 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
103 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F()
104 std::vector<TensorInfo> outputs = log_softmax->outputs_tensor_info(); in TEST_F()
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
110 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | pow_info_test.cc | 81 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F()
82 std::vector<TensorInfo> outputs = pow->outputs_tensor_info(); in TEST_F()
87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
88 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
102 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> outputs = pow->outputs_tensor_info(); in TEST_F()
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
109 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | gelu_info_test.cc | 81 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F()
82 std::vector<TensorInfo> outputs = gelu->outputs_tensor_info(); in TEST_F()
87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
88 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
102 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> outputs = gelu->outputs_tensor_info(); in TEST_F()
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
109 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | softmax_info_test.cc | 86 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F()
87 std::vector<TensorInfo> outputs = softmax->outputs_tensor_info(); in TEST_F()
92 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
93 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
107 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F()
108 std::vector<TensorInfo> outputs = softmax->outputs_tensor_info(); in TEST_F()
113 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
114 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | activation_info_test.cc | 82 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F()
83 std::vector<TensorInfo> outputs = activation->outputs_tensor_info(); in TEST_F()
88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
89 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
103 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F()
104 std::vector<TensorInfo> outputs = activation->outputs_tensor_info(); in TEST_F()
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
110 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | l2_normalize_info_test.cc | 82 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F()
83 std::vector<TensorInfo> outputs = norm->outputs_tensor_info(); in TEST_F()
88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
89 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
103 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F()
104 std::vector<TensorInfo> outputs = norm->outputs_tensor_info(); in TEST_F()
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
110 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | transpose_test.cc | 97 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F()
98 std::vector<TensorInfo> outputs = transpose->outputs_tensor_info(); in TEST_F()
103 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
104 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
118 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F()
119 std::vector<TensorInfo> outputs = transpose->outputs_tensor_info(); in TEST_F()
124 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
125 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| D | reduce_method_test.cc | 86 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F()
87 std::vector<TensorInfo> outputs = reduce_sum->outputs_tensor_info(); in TEST_F()
92 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
93 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
107 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F()
108 std::vector<TensorInfo> outputs = reduce_sum->outputs_tensor_info(); in TEST_F()
113 TensorInfo input_tensor_info = inputs.at(0); in TEST_F()
114 TensorInfo output_tensor_info = outputs.at(0); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | fullconnection_int8_tests.cc | 32 struct TensorInfo { struct
52 …TestInit(std::vector<lite::Tensor *> *inputs, std::vector<lite::Tensor *> *outputs, TensorInfo *in, in FcInt8TestInit()
53 TensorInfo *weight, TensorInfo *bias, TensorInfo *out) { in FcInt8TestInit()
82 TensorInfo in_params; in TEST_F()
101 TensorInfo weight_params; in TEST_F()
110 TensorInfo bias_params; in TEST_F()
120 TensorInfo out_params; in TEST_F()
|
| D | matmul_int8_tests.cc | 33 struct TensorInfo { struct
63 …TestInit(std::vector<lite::Tensor *> *inputs, std::vector<lite::Tensor *> *outputs, TensorInfo *in, in MMInt8TestInit()
64 TensorInfo *weight, TensorInfo *out) { in MMInt8TestInit()
87 TensorInfo in_params; in TEST_F()
101 TensorInfo weight_params; in TEST_F()
110 TensorInfo out_params; in TEST_F()
169 TensorInfo in_params; in TEST_F()
212 TensorInfo weight_params; in TEST_F()
228 TensorInfo out_params; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_info.h | 33 class TensorInfo {
35 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo() function
37 explicit TensorInfo(const TensorLayout &tensor_layout) : tensor_layout_(tensor_layout) { in TensorInfo() function
42 TensorInfo() = default;
43 ~TensorInfo() = default;
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 64 TensorInfo input0(input0_layout, input0_shape, input0_slice_shape), in TEST_F()
68 std::vector<TensorInfo> inputs, outputs; in TEST_F()
121 TensorInfo input0_info(input0_layout, input0_shape, input0_slice_shape), in TEST_F()
123 std::vector<TensorInfo> inputs, outputs; in TEST_F()
174 TensorInfo input_info(input_layout, input_shape, input_slice_shape); in TEST_F()
175 TensorInfo param_info(param_layout, param_shape, param_slice_shape); in TEST_F()
176 TensorInfo output_info(output_layout, output_shape, output_slice_shape); in TEST_F()
177 std::vector<TensorInfo> inputs, outputs; in TEST_F()
|
| /third_party/mindspore/tests/ut/python/debugger/gpu_tests/ |
| D | test_read_tensors_nonexist_node.py | 40 info1 = d.TensorInfo(node_name="Default/CudnnUniformReal-op391",
44 info2 = d.TensorInfo(node_name="Gradients/Default/network-WithLossCell/_backbone-AlexNet"
64 info_nonexist = d.TensorInfo(node_name="Default/CudnnUniformReal-op390",
82 … info_nonexist = d.TensorInfo(node_name="Gradients/Default/network-WithLossCell/_backbone-AlexNet/"
100 … info_nonexist = d.TensorInfo(node_name="Gradients/Default/network-WithLossCell/_backbone-AlexNet/"
|