Searched refs:remaining_input_shape (Results 1 – 1 of 1) sorted by relevance
2298 ShapeHandle remaining_input_shape; in SpaceToBatchShapeHelper() local2300 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()2304 remaining_input_shape, &result)); in SpaceToBatchShapeHelper()2373 ShapeHandle remaining_input_shape; in BatchToSpaceShapeHelper() local2375 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()2379 remaining_input_shape, &result)); in BatchToSpaceShapeHelper()