Searched refs:cropped_size (Results 1 – 2 of 2) sorted by relevance
156 const int64 cropped_size = in BatchToSpaceOpCompute() local158 OP_REQUIRES(context, cropped_size >= 0, in BatchToSpaceOpCompute()160 cropped_size, " must be non-negative")); in BatchToSpaceOpCompute()162 internal_output_shape.AddDim(cropped_size); in BatchToSpaceOpCompute()163 external_output_shape.AddDim(cropped_size); in BatchToSpaceOpCompute()
2205 DimensionHandle cropped_size; in BatchToSpaceShapeHelper() local2207 block_shape_vec[dim], &cropped_size)); in BatchToSpaceShapeHelper()2209 c->Subtract(cropped_size, crop_start, &cropped_size)); in BatchToSpaceShapeHelper()2211 c->Subtract(cropped_size, crop_end, &output_dims[dim + 1])); in BatchToSpaceShapeHelper()