Searched refs:output_shape_dims (Results 1 – 1 of 1) sorted by relevance
32 absl::Span<const int64> output_shape_dims, HloInstruction** param, in CreateModuleWithProgramShape() argument36 ShapeUtil::MakeShape(primitive_type, output_shape_dims); in CreateModuleWithProgramShape()