Searched defs:output_shape_dims (Results 1 – 3 of 3) sorted by relevance
32 absl::Span<const int64> output_shape_dims, HloInstruction** param, in CreateModuleWithProgramShape()47 absl::Span<const int64> output_shape_dims, HloInstruction** param, in CreateModuleWithProgramShape()
41 RaggedTensorToTensorOpModel(int output_shape_dims, in RaggedTensorToTensorOpModel()
2450 OpConverterParams* params, const int* output_shape_dims, in ConvertStaticReshapeForImplicitBatchMode()2517 const int* output_shape_dims = static_cast<int*>(weights.GetValues()); in ConvertReshape() local