Searched refs:output_shape_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv_test.cc | 58 std::initializer_list<int> output_shape_data, in BaseTransposeConvOpModel() argument
71 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in BaseTransposeConvOpModel()
88 PopulateTensor<int32_t>(output_shape_, output_shape_data); in BaseTransposeConvOpModel()
574 std::initializer_list<int> output_shape_data, in BaseTransposeConvBiasOpModel() argument
585 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in BaseTransposeConvBiasOpModel()
634 PopulateTensor<int32_t>(output_shape_, output_shape_data); in BaseTransposeConvBiasOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_conv_test.cc | 25 QuantizedTransposeConvOpModel(std::initializer_list<int> output_shape_data, in QuantizedTransposeConvOpModel() argument
35 output_shape_ = AddConstInput(TensorType_INT32, output_shape_data, {4}); in QuantizedTransposeConvOpModel()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1322 std::vector<int>* output_shape_data = output_shape.mutable_dims(); in ProcessSpaceToBatchNDOperator() local
1328 output_shape_data->at(dim + 1) = final_dim_size / block_shape_data[dim]; in ProcessSpaceToBatchNDOperator()
1332 output_shape_data->at(0) = output_batch_size; in ProcessSpaceToBatchNDOperator()
1333 output_shape_data->at(input_shape.dimensions_count() - 1) = in ProcessSpaceToBatchNDOperator()
1375 std::vector<int>* output_shape_data = output_shape.mutable_dims(); in ProcessBatchToSpaceNDOperator() local
1381 output_shape_data->at(dim + 1) = in ProcessBatchToSpaceNDOperator()
1385 output_shape_data->at(0) = output_batch_size; in ProcessBatchToSpaceNDOperator()
1386 output_shape_data->at(input_shape.dimensions_count() - 1) = in ProcessBatchToSpaceNDOperator()
1893 const std::vector<int64>& output_shape_data = in ProcessSparseToDenseOperator() local
1898 output_shape_data.begin(), output_shape_data.end(), in ProcessSparseToDenseOperator()
|