Searched refs:block_shape_array_shape (Results 1 – 1 of 1) sorted by relevance
1298 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessSpaceToBatchNDOperator() local1300 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessSpaceToBatchNDOperator()1304 QCHECK_EQ(block_shape_array_shape.dims(0), spatial_dims_num); in ProcessSpaceToBatchNDOperator()1352 const auto& block_shape_array_shape = block_shape_array.shape(); in ProcessBatchToSpaceNDOperator() local1354 QCHECK_EQ(block_shape_array_shape.dimensions_count(), 1); in ProcessBatchToSpaceNDOperator()1358 QCHECK_EQ(block_shape_array_shape.dims(0), spatial_dims_num); in ProcessBatchToSpaceNDOperator()