Searched refs:spatial_dims_num (Results 1 – 3 of 3) sorted by relevance
62 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local66 spatial_dims_num); in ResizeOutputTensor()70 spatial_dims_num); in ResizeOutputTensor()73 for (int i = 0; i < spatial_dims_num * 2; ++i) { in ResizeOutputTensor()79 for (int dim = 0; dim < spatial_dims_num; ++dim) { in ResizeOutputTensor()
63 int spatial_dims_num = input_size->size - 2; in ResizeOutputTensor() local67 spatial_dims_num); in ResizeOutputTensor()71 spatial_dims_num); in ResizeOutputTensor()79 for (int dim = 0; dim < spatial_dims_num; ++dim) { in ResizeOutputTensor()
1303 int spatial_dims_num = input_shape.dimensions_count() - 2; in ProcessSpaceToBatchNDOperator() local1304 QCHECK_EQ(block_shape_array_shape.dims(0), spatial_dims_num); in ProcessSpaceToBatchNDOperator()1312 QCHECK_EQ(paddings_array_shape.dims(0), spatial_dims_num); in ProcessSpaceToBatchNDOperator()1324 for (int dim = 0; dim < spatial_dims_num; ++dim) { in ProcessSpaceToBatchNDOperator()1348 int spatial_dims_num = input_shape.dimensions_count() - 2; in ProcessBatchToSpaceNDOperator() local1358 QCHECK_EQ(block_shape_array_shape.dims(0), spatial_dims_num); in ProcessBatchToSpaceNDOperator()1366 QCHECK_EQ(crops_array_shape.dims(0), spatial_dims_num); in ProcessBatchToSpaceNDOperator()1377 for (int dim = 0; dim < spatial_dims_num; ++dim) { in ProcessBatchToSpaceNDOperator()