Searched defs:expand_shape (Results 1 – 3 of 3) sorted by relevance
66 auto expand_shape = AnfAlgo::GetOutputInferShape(input_node, 0); in AddBroadCastToNode() local
92 Shape expand_shape = expand_list[i].GetFrontElementByValue(GetDimByIdx(i)); in GetExpandedShapeByExpandListRemoveLeft() local
174 ShapeVector expand_shape(data_info.shape_.begin(), data_info.shape_.end()); in ConvertTensorToNCOrNH() local