Searched refs:GetShapeInfo (Results 1 – 1 of 1) sorted by relevance
66 ShapeInfo GetShapeInfo( in GetShapeInfo() function536 GetShapeInfo(conv->operand(0)->shape(), dim_nums.input_batch_dimension(), in EmitConvolutionForwardAsMlir()540 ShapeInfo filter_shape_info = GetShapeInfo( in EmitConvolutionForwardAsMlir()545 ShapeInfo output_shape_info = GetShapeInfo( in EmitConvolutionForwardAsMlir()