Searched refs:crop_end (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batchtospace_op_test.py | 269 def _compare(self, b, h, w, d, block_size, crop_beg, crop_end): argument
275 [[crop_beg, crop_end], [crop_beg, crop_end]], dtype=np.int32)
285 crop_end = 0
286 self._compare(1, 2, 3, 5, block_size, crop_beg, crop_end)
292 crop_end = 0
293 self._compare(2, 4, 3, 2, block_size, crop_beg, crop_end)
299 crop_end = 1
300 self._compare(1, 2, 3, 5, block_size, crop_beg, crop_end)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 130 int64 crop_end = crops.Get<int64>({i, 1}); in BatchToSpace() local
131 OP_REQUIRES(ctx, crop_start >= 0 && crop_end >= 0, in BatchToSpace()
134 end_indices[1 + i] -= crop_end; in BatchToSpace()
139 " end: ", crop_end, " size ", reshaped_permuted_shape[1 + i])); in BatchToSpace()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batchtospace_op.cc | 151 crop_end = crops[2 * block_dim + 1]; in BatchToSpaceOpCompute() local
152 OP_REQUIRES(context, crop_start >= 0 && crop_end >= 0, in BatchToSpaceOpCompute()
157 input_size * block_shape_value - crop_start - crop_end; in BatchToSpaceOpCompute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchToSpaceND.pbtxt | 20 `crops[i] = [crop_start, crop_end]` specifies the amount to crop from input
23 `crop_start[i] + crop_end[i] <= block_shape[i] * input_shape[i + 1]`.
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1162 int64_t crop_end = crop_values[i * 2 + 1].getSExtValue(); in matchAndRewrite() local
1164 if (crop_start < 0 || crop_end < 0) { in matchAndRewrite()
1170 slice_sizes[i + 1] -= crop_start + crop_end; in matchAndRewrite()
1174 << crop_start << " end: " << crop_end << " size " in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2357 crop_end = crops_vec[dim * 2 + 1]; in BatchToSpaceShapeHelper() local
2358 if (crop_start < 0 || crop_end < 0) { in BatchToSpaceShapeHelper()
2368 c->Subtract(cropped_size, crop_end, &output_dims[dim + 1])); in BatchToSpaceShapeHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1104 `crops[i] = [crop_start, crop_end]` specifies the amount to crop from input
1107 `crop_start[i] + crop_end[i] <= block_shape[i] * input_shape[i + 1]`.
|