Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc176 auto& specified_output_shape_array = in ProcessTransposeConvOperator() local
178 if (!specified_output_shape_array.has_shape() || in ProcessTransposeConvOperator()
179 !specified_output_shape_array.buffer) { in ProcessTransposeConvOperator()
184 CHECK(specified_output_shape_array.data_type == ArrayDataType::kInt32) in ProcessTransposeConvOperator()
187 CHECK(specified_output_shape_array.shape().dimensions_count() == 1 && in ProcessTransposeConvOperator()
188 specified_output_shape_array.shape().dims(0) == 4) in ProcessTransposeConvOperator()
192 << toco::ShapeToString(specified_output_shape_array.shape()); in ProcessTransposeConvOperator()
239 specified_output_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTransposeConvOperator()