| /external/tensorflow/tensorflow/python/kernel_tests/ |
| batchtospace_op_test.py | 254 def _compare(self, b, h, w, d, block_size, crop_beg, crop_end):
260 [[crop_beg, crop_end], [crop_beg, crop_end]], dtype=np.int32)
269 crop_end = 0
270 self._compare(1, 2, 3, 5, block_size, crop_beg, crop_end)
275 crop_end = 0
276 self._compare(2, 4, 3, 2, block_size, crop_beg, crop_end)
281 crop_end = 1
282 self._compare(1, 2, 3, 5, block_size, crop_beg, crop_end)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batchtospace_op.cc | 131 int64 crop_end = crops.Get<int64>({i, 1}); local
132 OP_REQUIRES(ctx, crop_start >= 0 && crop_end >= 0,
135 end_indices[1 + i] -= crop_end;
140 " end: ", crop_end, " size ", reshaped_permuted_shape[1 + i]));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| batchtospace_op.cc | 151 crop_end = crops[2 * block_dim + 1]; local
152 OP_REQUIRES(context, crop_start >= 0 && crop_end >= 0,
157 input_size * block_shape_value - crop_start - crop_end;
|
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops.cc | 2055 crop_end = crops_vec[dim * 2 + 1]; local
[all...] |
