Home
last modified time | relevance | path

Searched refs:specified_output_shape_array (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc175 auto& specified_output_shape_array = in ProcessTransposeConvOperator() local
177 if (!specified_output_shape_array.has_shape() || in ProcessTransposeConvOperator()
178 !specified_output_shape_array.buffer) { in ProcessTransposeConvOperator()
183 CHECK(specified_output_shape_array.data_type == ArrayDataType::kInt32) in ProcessTransposeConvOperator()
186 CHECK(specified_output_shape_array.shape().dimensions_count() == 1 && in ProcessTransposeConvOperator()
187 specified_output_shape_array.shape().dims(0) == 4) in ProcessTransposeConvOperator()
191 << toco::ShapeToString(specified_output_shape_array.shape()); in ProcessTransposeConvOperator()
238 specified_output_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTransposeConvOperator()