Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc105 std::vector<int>* space_to_batch_output_dims) { in PopulateSpaceToBatchOutputDims() argument
107 space_to_batch_output_dims->push_back( in PopulateSpaceToBatchOutputDims()
110 space_to_batch_output_dims->push_back((space_to_batch_paddings[0] + in PopulateSpaceToBatchOutputDims()
115 space_to_batch_output_dims->push_back((space_to_batch_paddings[2] + in PopulateSpaceToBatchOutputDims()
120 space_to_batch_output_dims->push_back(input_depth_size); in PopulateSpaceToBatchOutputDims()
168 std::vector<int> space_to_batch_output_dims; in BuildDilatedDwConv() local
172 &space_to_batch_output_dims); in BuildDilatedDwConv()
174 AddOutput(sizeof(uint8_t), 4, space_to_batch_output_dims); in BuildDilatedDwConv()