Searched refs:output_batch_size (Results 1 – 6 of 6) sorted by relevance
78 const int output_batch_size = in ResizeOutputTensor() local93 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
80 const int output_batch_size = in ResizeOutputTensor() local84 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
3096 const int output_batch_size = output_shape.Dims(0); in SpaceToBatchND() local3106 for (int out_b = 0; out_b < output_batch_size; ++out_b) { in SpaceToBatchND()3149 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local3162 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()3163 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
3593 const int64_t output_batch_size =3625 uint8* output_ptr = output_data + b * output_batch_size;
5795 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local5808 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()5809 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
6123 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local6136 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()6137 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()