Searched refs:pad_beg (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | spacetobatch_op_test.py | 545 def _compare(self, b, h, w, d, block_size, pad_beg, pad_end): argument
551 [[pad_beg, pad_end], [pad_beg, pad_end]], dtype=np.int32)
560 pad_beg = 0
562 self._compare(1, 2, 3, 5, block_size, pad_beg, pad_end)
567 pad_beg = 0
569 self._compare(2, 4, 3, 2, block_size, pad_beg, pad_end)
574 pad_beg = 1
576 self._compare(1, 2, 3, 5, block_size, pad_beg, pad_end)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 169 int32_t pad_beg = (pad_total / 2 + 1) / block_size; in HandlePad() local
171 SmallVector<int32_t, 8> values = {0, 0, pad_beg, pad_end, in HandlePad()
172 pad_beg, pad_end, 0, 0}; in HandlePad()
|