Searched refs:remaining_input_shape (Results 1 – 1 of 1) sorted by relevance
2141 ShapeHandle remaining_input_shape; in SpaceToBatchShapeHelper() local2143 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()2147 remaining_input_shape, &result)); in SpaceToBatchShapeHelper()2216 ShapeHandle remaining_input_shape; in BatchToSpaceShapeHelper() local2218 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()2222 remaining_input_shape, &result)); in BatchToSpaceShapeHelper()