Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.cc34 if (inputs_shape_.size() != GATHER_V2_INPUTS_SIZE) { in GetAttrs()
35 MS_LOG(ERROR) << name_ << ": inputs shape size must be 2, but is " << inputs_shape_.size(); in GetAttrs()
54 if (inputs_shape_.at(0).size() == 0) { in GetAttrs()
59 …if (axis >= SizeToLong(inputs_shape_.at(0).size()) || axis < -SizeToLong(inputs_shape_.at(0).size(… in GetAttrs()
60 MS_LOG(ERROR) << "Axis is " << axis << ", not in [-" << inputs_shape_.at(0).size() << ", " in GetAttrs()
61 << inputs_shape_.at(0).size() << ")."; in GetAttrs()
64 axis += SizeToLong(inputs_shape_[0].size()); in GetAttrs()
68 index_size_ = inputs_shape_.at(1).size(); in GetAttrs()
74 if (inputs_shape_.size() != GATHER_V2_INPUTS_SIZE) { in CheckStrategy()
76 << inputs_shape_.size(); in CheckStrategy()
[all …]
Dselect_info.cc33 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
38 if (inputs_shape_.size() != 3) { in CheckStrategy()
43 if ((inputs_shape_[0] != inputs_shape_[1]) || (inputs_shape_[1] != inputs_shape_[2])) { in CheckStrategy()
76 if (inputs_shape_.empty()) { in InferTensorMap()
82 int64_t size = SizeToLong(inputs_shape_[0].size()); in InferTensorMap()
87 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in InferTensorMap()
95 for (size_t i = 0; i < inputs_shape_.size(); i++) { in ReComputeBatchSplitFlagList()
104 Shape input_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
106 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
120 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in GenerateOpStrategies()
Dprelu_info.cc37 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
46 if (stra[0][PRELU_CHANNEL_INDEX] != stra[1][0] && inputs_shape_[1][0] != 1) { in CheckStrategy()
70 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in InferTensorMap()
71 input_tensor_map.push_back((int64_t)(inputs_shape_[0].size() - i - 1)); in InferTensorMap()
75 if (inputs_shape_[1][0] == 1) { in InferTensorMap()
87 …if ((inputs_shape_.size() != PRELU_INPUTS_SIZE) || (outputs_shape_.size() != PRELU_OUTPUTS_SIZE)) { in GetAttrs()
88 …MS_LOG(ERROR) << name_ << ": Inputs shape size " << inputs_shape_.size() << " or outputs shape siz… in GetAttrs()
118 (void)input0_split.insert(input0_split.end(), inputs_shape_[0].size() - 2, 1); in GenerateOpStrategies()
119 Shape input1_split(inputs_shape_[1].size(), 0); in GenerateOpStrategies()
122 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dconcat_info.cc47 if (inputs_shape_.empty()) { in GetAttrs()
51 int64_t dim = SizeToLong(inputs_shape_[0].size()); in GetAttrs()
63 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
74 if (stra.size() != inputs_shape_.size()) { in CheckStrategy()
81 auto input_shape_ele = inputs_shape_[i]; in CheckStrategy()
122 if (inputs_shape_.empty()) { in InferTensorMap()
128 int64_t size = SizeToLong(inputs_shape_[0].size()); in InferTensorMap()
133 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in InferTensorMap()
141 for (size_t i = 0; i < inputs_shape_.size(); i++) { in ReComputeBatchSplitFlagList()
149 if (inputs_shape_.empty()) { in GenerateOpStrategies()
[all …]
Dpack_info.cc47 if (inputs_shape_.empty()) { in GetAttrs()
51 int64_t dim = SizeToLong(inputs_shape_[0].size()); in GetAttrs()
62 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
102 if (inputs_shape_.empty()) { in InferTensorMap()
107 int32_t size = SizeToInt(inputs_shape_[0].size()); in InferTensorMap()
113 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in InferTensorMap()
123 for (size_t i = 0; i < inputs_shape_.size(); i++) { in ReComputeBatchSplitFlagList()
131 if (inputs_shape_.empty()) { in GenerateOpStrategies()
135 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in GenerateOpStrategies()
141 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
[all …]
Dunsorted_segment_op_info.cc34 if (inputs_shape_.size() != UNSORTEDSEGMENTOP_INPUTS_SIZE) { in GetAttrs()
35 MS_LOG(ERROR) << name_ << ": inputs shape size must be 2, but is " << inputs_shape_.size(); in GetAttrs()
47 if (inputs_shape_.at(0).empty()) { in GetAttrs()
62 if (inputs_shape_.size() != UNSORTEDSEGMENTOP_INPUTS_SIZE) { in CheckStrategy()
64 << inputs_shape_.size(); in CheckStrategy()
73 if (CheckStrategyValue(strategy, {inputs_shape_.at(0), inputs_shape_.at(1)}) != SUCCESS) { in CheckStrategy()
80 Shape input_a_shape = inputs_shape_.at(0); in CheckStrategy()
81 Shape input_b_shape = inputs_shape_.at(1); in CheckStrategy()
138 size_t input0_size = inputs_shape_.at(0).size(); in InferTensorMap()
147 … tensor_map_out.begin() + static_cast<different_type>(inputs_shape_.at(1).size() - 1)); in InferTensorMap()
[all …]
Dactivation_info.cc35 …:CheckStrategy(const StrategyPtr &strategy) { return CheckStrategyValue(strategy, inputs_shape_); } in CheckStrategy()
43 …if ((inputs_shape_.size() != ACTIVATION_INPUTS_SIZE) || (outputs_shape_.size() != ACTIVATION_OUTPU… in GetAttrs()
44 …MS_LOG(ERROR) << name_ << " : Inputs shape size(" << inputs_shape_.size() << ") or outputs shape s… in GetAttrs()
68 …if ((inputs_shape_.size() != ACTIVATION_INPUTS_SIZE) || (outputs_shape_.size() != ACTIVATION_OUTPU… in GetAttrs()
69 …MS_LOG(ERROR) << name_ << " : Inputs shape size(" << inputs_shape_.size() << ") or outputs shape s… in GetAttrs()
78 …if ((inputs_shape_.size() != ACTIVATION_INPUTS_SIZE) || (outputs_shape_.size() != ACTIVATION_OUTPU… in GenerateOpStrategies()
79 …MS_LOG(EXCEPTION) << name_ << " : Inputs shape size(" << inputs_shape_.size() << ") or outputs sha… in GenerateOpStrategies()
83 Shape input0_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
86 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
94 Shape input0_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
[all …]
Dgathernd_info.cc36 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
84 if (inputs_shape_.size() != 2) { in InferTensorMap()
89 if (outputs_shape_.empty() || outputs_shape_[0].size() < (inputs_shape_[1].size() - 1)) { in InferTensorMap()
94 TensorMap input_tensor_map(inputs_shape_[0].size(), MAP_NONE); // the input can not split in InferTensorMap()
98 int64_t size = SizeToLong(inputs_shape_[1].size()); in InferTensorMap()
104 for (size_t i = 0; i < (inputs_shape_[1].size() - 1); ++i) { in InferTensorMap()
122 if (inputs_shape_.empty()) { in GenerateOpStrategies()
127 Shape input_split(inputs_shape_[1].size(), 1); in GenerateOpStrategies()
130 Shapes tmp_inputs_shape = {inputs_shape_[1]}; in GenerateOpStrategies()
144 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
Dbatch_parallel_info.cc30 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
58 if (need_replace_input_ && !inputs_shape_.empty()) { in InferDevMatrixShape()
59 replace_shape_ = inputs_shape_[0]; in InferDevMatrixShape()
86 for (size_t i = 0; i < inputs_shape_.size(); i++) { in InferTensorMap()
88 for (size_t j = 0; j < inputs_shape_[i].size(); ++j) { in InferTensorMap()
113 if (!inputs_shape_.empty()) { in GetAttrs()
124 inputs_shape_.push_back(GetValue<Shape>(shape_ptr)); in GetAttrs()
155 for (size_t i = 0; i < inputs_shape_.size(); i++) { in GenerateOpStrategies()
156 Shape temp(inputs_shape_[i].size(), 1); in GenerateOpStrategies()
169 for (size_t i = 0; i < inputs_shape_.size(); i++) { in ReComputeBatchSplitFlagList()
Ddsd_matmul_info.cc48 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
110 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in InferTensorMap()
112 input_tensor_map_w1.push_back((int64_t)(inputs_shape_[0].size() - i - 1)); in InferTensorMap()
119 for (size_t i = 0; i < inputs_shape_[1].size(); ++i) { in InferTensorMap()
121 input_tensor_map_w2.push_back((int64_t)(inputs_shape_[1].size() - i - 1)); in InferTensorMap()
128 for (size_t i = 0; i < inputs_shape_[2].size(); ++i) { in InferTensorMap()
130 input_tensor_map_v.push_back((int64_t)(inputs_shape_[2].size() + 2 - i)); in InferTensorMap()
152 …if ((inputs_shape_.size() != DSD_MATMUL_INPUTS_SIZE) || (outputs_shape_.size() != DSD_MATMUL_OUTPU… in GetAttrs()
153 …MS_LOG(ERROR) << name_ << ": Inputs shape size " << inputs_shape_.size() << " or outputs shape siz… in GetAttrs()
183 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
Dgatherd_info.cc43 int64_t input_dim = SizeToLong(inputs_shape_[0].size()); in GetAttrs()
61 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
101 size_t size = inputs_shape_[0].size(); in InferTensorMap()
113 …if (inputs_shape_.empty() || outputs_shape_.empty() || inputs_tensor_map_.empty() || outputs_tenso… in InferTensorInfo()
120 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
145 if (inputs_shape_.empty()) { in InferMirrorOps()
150 if (inputs_tensor_map_.size() != inputs_shape_.size()) { in InferMirrorOps()
187 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in ReComputeBatchSplitFlagList()
195 Shape input0_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
198 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
[all …]
Dtensordot_info.cc65 if ((axes_int_ < 0) || (IntToSize(axes_int_) > inputs_shape_[0].size()) || in GetAttrs()
66 (IntToSize(axes_int_) > inputs_shape_[1].size())) { in GetAttrs()
82 ele_var += SizeToInt(inputs_shape_[i].size()); in GetAttrs()
90 ele += SizeToInt(inputs_shape_[i].size()); in GetAttrs()
115 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
218 for (size_t i = 0; i < inputs_shape_[1].size(); i++) { in InferTensorMapAxesInt()
219 input_b_tensor_map.push_back((int64_t)(inputs_shape_[1].size() - i - 1)); in InferTensorMapAxesInt()
225 …output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size() - IntToSize(axes_i… in InferTensorMapAxesInt()
226 output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size())); in InferTensorMapAxesInt()
238 for (size_t i = 0; i < size - inputs_shape_[0].size(); ++i) { in InferTensorMapAxesTuple()
[all …]
Dbroadcast_to_info.cc63 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
69 auto in_shape = inputs_shape_.at(0); in CheckStrategy()
95 if (inputs_shape_.empty()) { in InferTensorMap()
100 int32_t size = SizeToInt(inputs_shape_[0].size()); in InferTensorMap()
106 size_t len_diff = outputs_shape_.at(0).size() - inputs_shape_.at(0).size(); in InferTensorMap()
118 if (inputs_shape_.empty()) { in GenerateOpStrategies()
122 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in GenerateOpStrategies()
123 if (inputs_shape_[0][i] == 1) { in GenerateOpStrategies()
132 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
146 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in GenerateOpStrategies()
Dtranspose_info.cc30 …:CheckStrategy(const StrategyPtr &strategy) { return CheckStrategyValue(strategy, inputs_shape_); } in CheckStrategy()
64 if (elements.size() != inputs_shape_[0].size()) { in ComputeAxis()
91 if ((inputs_shape_.size() != 1) || (outputs_shape_.size() != 1)) { in InferTensorMap()
93 << inputs_shape_.size() << ", " << outputs_shape_.size(); in InferTensorMap()
98 for (size_t j = 0; j < inputs_shape_[0].size(); ++j) { in InferTensorMap()
99 tensor_map_index_input.push_back(SizeToLong(inputs_shape_[0].size() - j - 1)); in InferTensorMap()
138 Shape input0_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
141 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dloss_info.cc31 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
46 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy()
63 if ((inputs_shape_.size() != SoftmaxCrossEntropyWithLogitsInputsSize) || in GetAttrs()
81 size_t size = inputs_shape_[0].size(); in InferTensorMap()
129 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in ReComputeBatchSplitFlagList()
137 size_t input_dim = inputs_shape_[0].size(); in GenerateOpStrategies()
142 (void)input0_split.insert(input0_split.begin(), inputs_shape_[0].size(), 1); in GenerateOpStrategies()
146 …if (GenerateStrategiesWithBroadcast(stage_id, inputs_shape_, splittable_inputs, &sp_vector) != SUC… in GenerateOpStrategies()
Dsplit_info.cc49 if (inputs_shape_.empty()) { in GetAttrs()
53 int dim = SizeToInt(inputs_shape_[0].size()); in GetAttrs()
79 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
117 if (inputs_shape_.empty()) { in InferTensorMap()
122 int32_t size = SizeToInt(inputs_shape_[0].size()); in InferTensorMap()
140 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in GenerateOpStrategies()
149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
163 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies()
165 if (inputs_shape_[0].size() > 1) { in GenerateBatchStrategies()
Dtmp_identity_info.cc28 return CheckStrategyValue(strategy, inputs_shape_); in CheckStrategy()
40 size_t size = inputs_shape_[0].size(); in InferTensorMap()
74 if ((inputs_shape_.size() != 1) || (outputs_shape_.size() != 1)) { in GenerateOpStrategies()
75 … << name_ << ": Inputs shape size or outputs shape size is wrong, " << inputs_shape_.size() << ", " in GenerateOpStrategies()
78 Shape input0_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
81 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dmatmul_dds_info.cc47 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
136 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in InferTensorMap()
138 input_tensor_map_q.push_back((int64_t)(inputs_shape_[0].size() + 3 - i - 1)); in InferTensorMap()
145 for (size_t i = 0; i < inputs_shape_[1].size(); ++i) { in InferTensorMap()
147 input_tensor_map_k.push_back((int64_t)(inputs_shape_[1].size() + 3 - i - 1)); in InferTensorMap()
154 for (size_t i = 0; i < inputs_shape_[2].size(); ++i) { in InferTensorMap()
156 input_tensor_map_local_mask.push_back((int64_t)(inputs_shape_[2].size() + 3 - 2)); in InferTensorMap()
163 for (size_t i = 0; i < inputs_shape_[3].size(); ++i) { in InferTensorMap()
165 input_tensor_map_global_mask.push_back((int64_t)(inputs_shape_[3].size() + 3 - 2)); in InferTensorMap()
202 …if ((inputs_shape_.size() != MATMUL_DDS_INPUTS_SIZE) || (outputs_shape_.size() != MATMUL_DDS_OUTPU… in GetAttrs()
[all …]
Dtopk_info.cc33 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
66 if (inputs_shape_.empty()) { in InferTensorMap()
72 int64_t size = SizeToLong(inputs_shape_[0].size()); in InferTensorMap()
77 for (size_t i = 0; i < inputs_shape_.size(); ++i) { in InferTensorMap()
109 if (inputs_shape_.empty()) { in InferMirrorOps()
114 if (inputs_tensor_map_.size() != inputs_shape_.size()) { in InferMirrorOps()
155 Shape input_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
158 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
Dscatter_update_info.cc42 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
107 if (inputs_shape_.size() != 3) { in InferTensorMap()
113 TensorMap indices_tensor_map(inputs_shape_[1].size(), MAP_NONE); in InferTensorMap()
116 int64_t size = SizeToLong(inputs_shape_[0].size()); in InferTensorMap()
135 for (size_t i = 0; i < inputs_shape_.size(); i++) { in ReComputeBatchSplitFlagList()
146 Shape input_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies()
163 Dimensions indices_strategy(inputs_shape_[1].size(), 1); in GenerateOpStrategies()
Dl2_normalize_info.cc30 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
38 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy()
66 Shape input0_split(inputs_shape_[0].size() - 1, 1); in GenerateOpStrategies()
69 size_t input_dim = inputs_shape_.at(0).size(); in GenerateOpStrategies()
76 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dgather_v2_p_info.cc130 auto params_shape = inputs_shape_.at(0); in GetAttrs()
136 axis += SizeToLong(inputs_shape_[0].size()); in GetAttrs()
160 if (std::find(inputs_shape_[1].begin(), inputs_shape_[1].end(), -1) != inputs_shape_[1].end()) { in GetAttrs()
200 int64_t min_param_slice_row = inputs_shape_[1][1] / indices_strategy[1]; in CheckManualSplit()
208 if (inputs_shape_[0][0] < inputs_shape_[1][1]) { in CheckManualSplit()
222 if (split_shape_sum != inputs_shape_[0][0]) { in CheckManualSplit()
291 Shape index_shape = inputs_shape_.at(1); in SetAttribute()
316 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
321 auto param_shape = inputs_shape_.at(0); in CheckStrategy()
330 if (inputs_shape_.at(0).size() != 1 && inputs_shape_.at(0).size() != 2) { in CheckStrategy()
[all …]
Dstrided_slice_info.cc101 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
137 bool no_fully_fetch = ((begin_[i] != 0) || (end_[i] < inputs_shape_[0][i])); in CheckStrategy()
161 if (inputs_shape_.empty()) { in InferTensorMap()
167 int64_t size = SizeToLong(inputs_shape_[0].size()); in InferTensorMap()
207 return GenerateBatchStrategiesBySplitFlag(inputs_shape_, split_flag_list_); in GenerateBatchStrategies()
215 Shape input_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
217 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in GenerateOpStrategies()
222 bool no_fully_fetch = ((begin_[i] != 0) || (end_[i] < inputs_shape_[0][i])); in GenerateOpStrategies()
231 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dslice_info.cc71 if (CheckStrategyValue(strategy, inputs_shape_) != SUCCESS) { in CheckStrategy()
85 bool no_fully_fetch = ((begin_[i] != 0) || (size_[i] < inputs_shape_[0][i])); in CheckStrategy()
109 if (inputs_shape_.empty()) { in InferTensorMap()
115 int64_t size = SizeToInt(inputs_shape_[0].size()); in InferTensorMap()
154 return GenerateBatchStrategiesBySplitFlag(inputs_shape_, split_flag_list_); in GenerateBatchStrategies()
160 Shape input_split(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
162 bool no_fully_fetch = ((begin_[i] != 0) || (size_[i] < inputs_shape_[0][i])); in GenerateOpStrategies()
170 …if (GenerateStrategiesForIndependentInputs(stage_id, inputs_shape_, splittable_inputs, &sp_vector)… in GenerateOpStrategies()
Dtile_info.cc75 tmp.push_back(inputs_shape_[0][i]); in CheckStrategy()
79 …MS_LOG(INFO) << name_ << ": The input shape is " << ShapeToString(inputs_shape_[0]) << ", the mult… in CheckStrategy()
111 if (inputs_shape_.empty() || outputs_shape_.empty()) { in InferTensorMap()
117 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in InferTensorMap()
118 input_tensor_map.push_back(inputs_shape_[0].size() - i - 1); in InferTensorMap()
120 for (size_t i = 0; i < inputs_shape_[0].size(); ++i) { in InferTensorMap()
201 tmp_input_shape[i] = inputs_shape_[0][i]; in GenerateOpStrategies()

12