Searched refs:outputs_layout (Results 1 – 7 of 7) sorted by relevance
56 Status GetNextInfo::InferTensorLayout(TensorLayouts *outputs_layout) { in InferTensorLayout() argument57 if (outputs_layout == nullptr) { in InferTensorLayout()66 outputs_layout->push_back(output_layout); in InferTensorLayout()72 TensorLayouts outputs_layout; in InferTensorInfo() local73 if (InferTensorLayout(&outputs_layout) != SUCCESS) { in InferTensorInfo()77 TensorInfo output_tensor_info(outputs_layout[i]); in InferTensorInfo()
192 Status ReshapeInfo::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout() argument193 if (inputs_layout == nullptr || outputs_layout == nullptr) { in InferTensorLayout()234 outputs_layout->push_back(tensor_layout_out); in InferTensorLayout()262 TensorLayouts inputs_layout, outputs_layout; in InferTensorInfo() local263 if (InferTensorLayout(&inputs_layout, &outputs_layout) != SUCCESS) { in InferTensorInfo()267 TensorLayout tensor_layout_out = outputs_layout.at(0); in InferTensorInfo()
314 Status MatMulBase::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout() argument339 outputs_layout->push_back(output_layout); in InferTensorLayout()345 TensorLayouts inputs_layout, outputs_layout; in InferTensorInfo() local346 if (InferTensorLayout(&inputs_layout, &outputs_layout) != SUCCESS) { in InferTensorInfo()352 TensorLayout output_layout = outputs_layout.at(0); in InferTensorInfo()
48 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);
47 Status InferTensorLayout(TensorLayouts *outputs_layout);
57 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);
80 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);