Home
last modified time | relevance | path

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

/third_party/mindspore/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
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
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
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
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
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
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
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
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
Dreduce_method_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
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgatherd_info.cc124 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
Dgather_v2_info.cc194 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
Dreduce_method_info.cc371 TensorInfo input_tensor_info(input_tensor_layout, input_shape, input_slice_shape); in InferTensorInfo() local
500 TensorInfo input_tensor_info(input_tensor_layout, input_shape, input_slice_shape); in InferTensorInfo() local
Dgather_v2_p_info.cc586 TensorInfo input_tensor_info(input_tensor_layout); in InferTensorInfo() local
Doperator_info.cc214 TensorInfo input_tensor_info(input_layout); in InferTensorInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc463 InputTensorInfo *const input_tensor_info) { in GetSingleOpInputTensors()
472InputTensorInfo *const input_tensor_info, size_t input_index) { in GetSingleOpInputTensorByIndex()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc513 … const std::vector<tensor::TensorPtr> &graph_inputs, InputTensorInfo *input_tensor_info, in GetControlOpInput()
603 … const std::vector<tensor::TensorPtr> &graph_inputs, InputTensorInfo *input_tensor_info, in RunControlOperator()
741 InputTensorInfo input_tensor_info; in RunGraphBySingleOp() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc1083 InputTensorInfo *input_tensor_info) { in GetOpInputStubTensors()
1131 InputTensorInfo input_tensor_info; in BuildOpsInGraph() local
Dsession_basic.cc1436 InputTensorInfo *input_tensor_info) { in GetOpInputTensors()
1472InputTensorInfo *const input_tensor_info, size_t input_index) { in GetOpInputTensorByIndex()
2293 InputTensorInfo input_tensor_info; in RunOpsInGraphImpl() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc1119 TensorInfo input_tensor_info = inputs[0]; in GetBackwardCommCost() local