Searched refs:block_shape_shape (Results 1 – 1 of 1) sorted by relevance
2234 ShapeHandle block_shape_shape, in SpaceToBatchShapeHelper() argument2238 if (c->Rank(block_shape_shape) != 1) { in SpaceToBatchShapeHelper()2242 const DimensionHandle num_block_dims_handle = c->Dim(block_shape_shape, 0); in SpaceToBatchShapeHelper()2310 ShapeHandle block_shape_shape, in BatchToSpaceShapeHelper() argument2313 if (c->Rank(block_shape_shape) != 1) { in BatchToSpaceShapeHelper()2317 const DimensionHandle num_block_dims_handle = c->Dim(block_shape_shape, 0); in BatchToSpaceShapeHelper()