Searched defs:output_layout (Results 1 – 12 of 12) sorted by relevance
89 TensorLayout output_layout; in InferTensorInfo() local
54 void SetOutputLayout(const TensorLayout &output_layout) { in SetOutputLayout()
131 TensorLayout output_layout; in InferTensorInfo() local
508 TensorLayout output_layout; in InferTensorLayout() local531 TensorLayout output_layout = outputs_layout.at(kOutput); in InferTensorInfo() local
514 TensorLayout output_layout; in InferTensorLayout() local530 TensorLayout output_layout = outputs_layout.at(kQbmmOutput); in InferTensorInfo() local
68 TensorLayout output_layout; in InferTensorLayout() local
165 TensorLayout output_layout; in InferTensorInfo() local
826 int64_t ReshapeInfo::GetSWCIndexByOutputLayoutWithZeroComm(const TensorLayout &output_layout) { in GetSWCIndexByOutputLayoutWithZeroComm()848 int64_t ReshapeInfo::GetSWCIndexByOutputLayoutWithMiniComm(const TensorLayout &output_layout) { in GetSWCIndexByOutputLayoutWithMiniComm()
932 TensorLayout output_layout; in InferTensorLayout() local958 TensorLayout output_layout = outputs_layout.at(0); in InferTensorInfo() local
642 TensorLayout output_layout; in InferTensorInfo() local1517 for (const auto &output_layout : out_tensor_layouts) { in InitWithTensorLayout() local2254 StrategyPtr OperatorInfo::GetStrategyFromSWCByOutputLayout(const TensorLayout &output_layout, size_… in GetStrategyFromSWCByOutputLayout()
167 TensorLayout input_layout, param_layout, output_layout; in TEST_F() local
62 AVChannelLayout output_layout; member