Searched defs:output_shape_dims (Results 1 – 4 of 4) sorted by relevance
34 absl::Span<const int64_t> output_shape_dims, HloInstruction** param, in CreateModuleWithProgramShape()49 absl::Span<const int64_t> output_shape_dims, HloInstruction** param, in CreateModuleWithProgramShape()
41 RaggedTensorToTensorOpModel(int output_shape_dims, in RaggedTensorToTensorOpModel()
2270 DimsAdapter output_shape_dims( in ConvertReshape() local5633 nvinfer1::Dims output_shape_dims; in ConvertResize() local
4593 const int output_shape_dims = SizeOfDimension(&output_shape_tensor, 0); in VisitTransposeConvNode() local