Searched refs:crop_start (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 129 int64 crop_start = crops.Get<int64>({i, 0}); in BatchToSpace() local
131 OP_REQUIRES(ctx, crop_start >= 0 && crop_end >= 0, in BatchToSpace()
133 start_indices[1 + i] = crop_start; in BatchToSpace()
138 "Cropped size must be non-negative: start: ", crop_start, in BatchToSpace()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batchtospace_op.cc | 150 const int64 crop_start = crops[2 * block_dim], 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 | 1161 int64_t crop_start = crop_values[i * 2].getSExtValue(); in matchAndRewrite() local
1164 if (crop_start < 0 || crop_end < 0) { in matchAndRewrite()
1169 start_indices[i + 1] = crop_start; 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 | 2356 const int64 crop_start = crops_vec[dim * 2], in BatchToSpaceShapeHelper() local
2358 if (crop_start < 0 || crop_end < 0) { in BatchToSpaceShapeHelper()
2366 c->Subtract(cropped_size, crop_start, &cropped_size)); 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]`.
|