Searched refs:output_shape_array (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | range.cc | 77 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(1); in ResizeOutput() local
78 output_shape_array->data[0] = size; in ResizeOutput()
79 return context->ResizeTensor(context, output, output_shape_array); in ResizeOutput()
|
| H A D | sparse_to_dense.cc | 43 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(output_dimensions); in Resize() local
45 output_shape_array->data[i] = GetTensorData<T>(output_shape)[i]; in Resize()
48 return context->ResizeTensor(context, output, output_shape_array); in Resize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_beam_search_decoder.cc | 113 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(dimensions); in Resize() local
116 output_shape_array->data[i++] = v; in Resize()
118 return context->ResizeTensor(context, output, output_shape_array); in Resize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tflite_op_shim.cc | 104 TfLiteIntArray* output_shape_array = in GetOutput() local
106 context_->ResizeTensor(context_, tflite_tensor, output_shape_array); in GetOutput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 1877 const Array& output_shape_array = model->GetArray(op->inputs[1]); in ProcessSparseToDenseOperator() local
1878 if (!output_shape_array.has_shape()) return; in ProcessSparseToDenseOperator()
1879 CHECK_EQ(output_shape_array.shape().dimensions_count(), 1); in ProcessSparseToDenseOperator()
1882 CHECK_LE(output_shape_array.shape().dims(0), 4); in ProcessSparseToDenseOperator()
1888 CHECK(output_shape_array.data_type == ArrayDataType::kInt32 || in ProcessSparseToDenseOperator()
1889 output_shape_array.data_type == ArrayDataType::kInt64); in ProcessSparseToDenseOperator()
1890 if (output_shape_array.data_type == ArrayDataType::kInt32) { in ProcessSparseToDenseOperator()
1892 output_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessSparseToDenseOperator()
1895 output_shape_array.GetBuffer<ArrayDataType::kInt64>().data; in ProcessSparseToDenseOperator()
|