Searched refs:block_shape_value (Results 1 – 5 of 5) sorted by relevance
154 const int64 block_shape_value = block_shape[block_dim]; in SpaceToBatchOpCompute() local156 if (padded_size % block_shape_value != 0) { in SpaceToBatchOpCompute()160 block_dim, "]=", block_shape_value); in SpaceToBatchOpCompute()163 const int64 output_size = padded_size / block_shape_value; in SpaceToBatchOpCompute()
155 const int64 block_shape_value = block_shape[block_dim]; in BatchToSpaceOpCompute() local157 input_size * block_shape_value - crop_start - crop_end; in BatchToSpaceOpCompute()
54 for block_dim, block_shape_value in enumerate(block_shape):55 reduced_size = padded.shape[block_dim + 1] // block_shape_value58 reshaped_padded_shape.append(block_shape_value)
58 for block_dim, block_shape_value in enumerate(block_shape):59 reduced_size = padded.shape[block_dim + 1] // block_shape_value62 reshaped_padded_shape.append(block_shape_value)
2103 const int64 block_shape_value = block_shape_vec[dim]; in SpaceToBatchShapeHelper() local2104 if (block_shape_value < 1) { in SpaceToBatchShapeHelper()2109 c->Multiply(batch_size, block_shape_value, &batch_size)); in SpaceToBatchShapeHelper()2178 const int64 block_shape_value = block_shape_vec[dim]; in BatchToSpaceShapeHelper() local2179 if (block_shape_value < 1) { in BatchToSpaceShapeHelper()2183 TF_RETURN_IF_ERROR(c->Divide(batch_size, block_shape_value, in BatchToSpaceShapeHelper()