Searched refs:space_to_batch_output_dims (Results 1 – 1 of 1) sorted by relevance
105 std::vector<int>* space_to_batch_output_dims) { in PopulateSpaceToBatchOutputDims() argument107 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() local172 &space_to_batch_output_dims); in BuildDilatedDwConv()174 AddOutput(sizeof(uint8_t), 4, space_to_batch_output_dims); in BuildDilatedDwConv()