Home
last modified time | relevance | path

Searched refs:inputs_tensor_info (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_parse_graph.cc85 size_t input_tensor_size = ops[iter_ops]->inputs_tensor_info().size(); in CompleteOperatorInputs()
94 if (ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape().size() == 4) { in CompleteOperatorInputs()
96 MakeTensor(ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[0], in CompleteOperatorInputs()
97 ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[1], in CompleteOperatorInputs()
98 ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[2], in CompleteOperatorInputs()
99 ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[3]); in CompleteOperatorInputs()
100 } else if (ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape().size() == 3) { in CompleteOperatorInputs()
102 MakeTensor(1, ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[0], in CompleteOperatorInputs()
103 ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[1], in CompleteOperatorInputs()
104 ops[iter_ops]->inputs_tensor_info()[iter_input_tensors].shape()[2]); in CompleteOperatorInputs()
[all …]
Drec_generate_strategy.cc85 auto shape_1 = ops[iter_ops]->inputs_tensor_info()[0].shape()[0]; in PrepareMatMul()
87 shape_1 = ops[iter_ops]->inputs_tensor_info()[0].shape()[1]; in PrepareMatMul()
89 auto shape_4 = ops[iter_ops]->inputs_tensor_info()[1].shape()[1]; in PrepareMatMul()
91 shape_4 = ops[iter_ops]->inputs_tensor_info()[1].shape()[0]; in PrepareMatMul()
124 …for (size_t iter_op_inputs = 0; iter_op_inputs < ops[iter_ops]->inputs_tensor_info().size(); iter_… in PrepareMatMul()
171 …bool no_fully_fetch = ((begin[i] != 0) || (end[i] < ops[iter_ops]->inputs_tensor_info()[0].shape()… in PrepareStridedSlice()
213 axis_input += SizeToLong(ops[iter_ops]->inputs_tensor_info()[0].shape().size()); in PrepareGatherV2()
256 axis_input += SizeToLong(ops[iter_ops]->inputs_tensor_info()[0].shape().size()); in PrepareGatherV2P()
265 for (size_t i = 1; i < ops[iter_ops]->inputs_tensor_info()[0].shape().size(); i++) { in PrepareGatherV2P()
266 s.push_back(strategie[ops[iter_ops]->inputs_tensor_info()[1].shape().size() - 1 + i]); in PrepareGatherV2P()
[all …]
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Donehot_info_test_axis_0.cc108 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
130 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
152 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
174 std::vector<TensorInfo> inputs = onehot_info2->inputs_tensor_info(); in TEST_F()
Dreshape_test.cc97 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
118 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
139 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
160 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F()
Donehot_info_test.cc108 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
130 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
152 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
174 std::vector<TensorInfo> inputs = onehot_info->inputs_tensor_info(); in TEST_F()
Dmatmul_info_test.cc171 std::vector<TensorInfo> inputs = matmul1->inputs_tensor_info(); in TEST_F()
197 std::vector<TensorInfo> inputs = matmul2->inputs_tensor_info(); in TEST_F()
223 std::vector<TensorInfo> inputs = matmul3->inputs_tensor_info(); in TEST_F()
248 std::vector<TensorInfo> inputs = matmul1->inputs_tensor_info(); in TEST_F()
274 std::vector<TensorInfo> inputs = matmul2->inputs_tensor_info(); in TEST_F()
300 std::vector<TensorInfo> inputs = matmul3->inputs_tensor_info(); in TEST_F()
326 std::vector<TensorInfo> inputs = matmul3->inputs_tensor_info(); in TEST_F()
570 std::vector<TensorInfo> inputs_info = matmul1->inputs_tensor_info(); in TEST_F()
587 std::vector<TensorInfo> inputs_info = matmul3->inputs_tensor_info(); in TEST_F()
Dtensor_add_info_test.cc84 std::vector<TensorInfo> inputs = tensor_add->inputs_tensor_info(); in TEST_F()
108 std::vector<TensorInfo> inputs = tensor_add->inputs_tensor_info(); in TEST_F()
189 std::vector<TensorInfo> inputs_info = tensor_add->inputs_tensor_info(); in TEST_F()
211 std::vector<TensorInfo> inputs_info = tensor_add1->inputs_tensor_info(); in TEST_F()
Dtmpidentity_test.cc83 std::vector<TensorInfo> inputs = identity_ptr->inputs_tensor_info(); in TEST_F()
104 std::vector<TensorInfo> inputs = identity_ptr->inputs_tensor_info(); in TEST_F()
146 std::vector<TensorInfo> inputs_info = identity_ptr->inputs_tensor_info(); in TEST_F()
Dprelu_test.cc82 std::vector<TensorInfo> inputs = prelu->inputs_tensor_info(); in TEST_F()
105 std::vector<TensorInfo> inputs = prelu->inputs_tensor_info(); in TEST_F()
187 std::vector<TensorInfo> inputs = prelu_2d->inputs_tensor_info(); in TEST_F()
210 std::vector<TensorInfo> inputs = prelu_2d->inputs_tensor_info(); in TEST_F()
Dactivation_test.cc85 std::vector<TensorInfo> inputs_info = act_ptr_->inputs_tensor_info(); in TEST_F()
110 std::vector<TensorInfo> inputs_info = soft_ptr_->inputs_tensor_info(); in TEST_F()
Dtanh_info_test.cc81 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F()
102 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F()
Dlog_softmax_info_test.cc82 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F()
Dpow_info_test.cc81 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F()
102 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F()
Dgelu_info_test.cc81 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F()
102 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F()
Dsoftmax_info_test.cc86 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F()
107 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F()
Dactivation_info_test.cc82 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F()
Dl2_normalize_info_test.cc82 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F()
103 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F()
Dsoftmax_entropy_loss_info_test.cc82 std::vector<TensorInfo> inputs = loss->inputs_tensor_info(); in TEST_F()
111 std::vector<TensorInfo> inputs = loss->inputs_tensor_info(); in TEST_F()
Dtranspose_test.cc97 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F()
118 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F()
Dreduce_method_test.cc86 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F()
107 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc456 if (LongToSize(index - 1) >= next_distribute_operator->inputs_tensor_info().size()) { in Redistribution()
458 … << next_distribute_operator->inputs_tensor_info().size() << "next operator name is " in Redistribution()
462 TensorInfo tensorinfo_out = next_distribute_operator->inputs_tensor_info()[LongToSize(index - 1)]; in Redistribution()
614 if (LongToSize(index - 1) >= op_info->inputs_tensor_info().size()) { in SplitTensor()
616 << op_info->inputs_tensor_info().size(); in SplitTensor()
618 TensorInfo tensor_info = op_info->inputs_tensor_info()[LongToSize(index - 1)]; in SplitTensor()
653 if (inputs_values.size() != op_info->inputs_tensor_info().size()) { in SplitTensorList()
655 << op_info->inputs_tensor_info().size(); in SplitTensorList()
668 TensorInfo tensor_info = op_info->inputs_tensor_info()[i]; in SplitTensorList()
1583 … } else if (IntToSize(param_pair.second - 1) >= distribute_operator->inputs_tensor_info().size()) { in ApplyParallelOptOnParam()
[all …]
Dparameter_manager.cc226 tensor_info = op_info->inputs_tensor_info()[param_index]; in GetParameterSliceInfo()
228 size_t input_tensor_info_size = op_info->inputs_tensor_info().size(); in GetParameterSliceInfo()
233 tensor_info = op_info->inputs_tensor_info()[LongToSize(user_input_index - 1)]; in GetParameterSliceInfo()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc579 tensor_info = op_info_pair.first->inputs_tensor_info().at(IntToSize(op_info_pair.second)); in InsertSend()
584 tensor_info = op_info_pair.first->inputs_tensor_info().at(IntToSize(op_info_pair.second)); in InsertSend()
640 tensor_info = op_info_pair.first->inputs_tensor_info().at(IntToSize(op_info_pair.second)); in InsertReceive()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h122 std::vector<TensorInfo> inputs_tensor_info() const { return inputs_tensor_info_; } in inputs_tensor_info() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dgraph_costmodel.cc1595 pre_info = pre_op_info->inputs_tensor_info()[LongToSize(pre_index)]; in InitReshapeStrategy()
1616 … (*next_iter)->next_operator()->inputs_tensor_info()[LongToSize(next_index)].tensor_layout()); in InitReshapeStrategy()