Searched refs:block_shape_shape (Results 1 – 1 of 1) sorted by relevance
2077 ShapeHandle block_shape_shape, in SpaceToBatchShapeHelper() argument2081 if (c->Rank(block_shape_shape) != 1) { in SpaceToBatchShapeHelper()2085 const DimensionHandle num_block_dims_handle = c->Dim(block_shape_shape, 0); in SpaceToBatchShapeHelper()2153 ShapeHandle block_shape_shape, in BatchToSpaceShapeHelper() argument2156 if (c->Rank(block_shape_shape) != 1) { in BatchToSpaceShapeHelper()2160 const DimensionHandle num_block_dims_handle = c->Dim(block_shape_shape, 0); in BatchToSpaceShapeHelper()