Searched defs:output_shape_dims (Results 1 – 4 of 4) sorted by relevance
113 TfLiteIntArray* output_shape_dims = IntArrayFromInts(output_shape_dims_data); in TestTransposeConvFloat() local158 TfLiteIntArray* output_shape_dims = IntArrayFromInts(output_shape_dims_data); in TestTransposeConvQuantized() local242 TfLiteIntArray* output_shape_dims = IntArrayFromInts(output_shape_dims_data); in TF_LITE_MICRO_TEST() local278 TfLiteIntArray* output_shape_dims = IntArrayFromInts(output_shape_dims_data); in TF_LITE_MICRO_TEST() local
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()
2587 const int* output_shape_dims = static_cast<int*>(weights.GetValues()); in ConvertReshape() local