Home
last modified time | relevance | path

Searched defs:input_tensor_info (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ops_info/
Donehot_info_test_axis_0.cc114 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
136 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
158 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
180 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Donehot_info_test.cc114 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
136 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
158 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
180 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dreshape_test.cc103 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
124 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
145 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
166 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dprelu_test.cc89 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
112 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
194 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
217 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dtanh_info_test.cc87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dgelu_info_test.cc87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dpow_info_test.cc87 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
108 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dtmpidentity_test.cc89 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
110 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dactivation_info_test.cc88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dlog_softmax_info_test.cc88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dtranspose_test.cc103 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
124 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dsoftmax_entropy_loss_info_test.cc90 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
119 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dl2_normalize_info_test.cc88 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
109 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dsoftmax_info_test.cc92 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
113 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
Dreduce_method_test.cc93 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
114 TensorInfo input_tensor_info = inputs.at(0); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dgrouped_matmul_info.cc263 auto input_tensor_info = CreateTensorInfo(device_matrix, shape, tensor_map); in CreateTensorInfo() local
270 TensorInfo input_tensor_info(input_layout); in CreateTensorInfo() local
288 auto input_tensor_info = in InferTensorInfo() local
Dstand_alone_info.cc80 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
Dreduce_method_info.cc297 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
378 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
591 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
Dgroup_norm_info.cc121 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
Dgatherd_info.cc157 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
Dgather_info.cc520 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
869 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
1315 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfoNoSplitAxis() local
Doperator_info.cc558 auto input_tensor_info = CreateTensorInfo(device_matrix, shape, tensor_map); in CreateTensorInfo() local
565 TensorInfo input_tensor_info(input_layout); in CreateTensorInfo() local
580 …auto input_tensor_info = CreateTensorInfo(dev_matrix_shape_, inputs_shape_new_[i], inputs_tensor_m… in InferTensorInfoNew() local
593 …f(inputs_tensor_info_.begin(), inputs_tensor_info_.end(), [](const TensorInfo &input_tensor_info) { in UpdateOutputTensorInfoForInterleaved()
636 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
1509 TensorInfo input_tensor_info(*input_layout); in InitWithTensorLayout() local
Dquant_batch_matmul_info.cc504 TensorInfo input_tensor_info(input_layout); in InferTensorLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc1202 TensorInfo input_tensor_info = inputs[0]; in GetBackwardCommCost() local
1993 TensorInfo input_tensor_info = inputs[i]; in GetBackwardCommCost() local