Searched refs:ExpandTensorShapeTo3dForPermute (Results 1 – 1 of 1) sorted by relevance
72 TensorShape ExpandTensorShapeTo3dForPermute(const TensorShape & inputShape) in ExpandTensorShapeTo3dForPermute() function216 newTensorInfo.SetShape(ExpandTensorShapeTo3dForPermute(tensorInfo.GetShape())); in PermuteInputsForConcat()234 ExpandTensorShapeTo3dForPermute(outputTensorInfo.GetShape()), in PermuteInputsForConcat()