Searched refs:base_paddings (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | spacetobatch_op_test.py | 629 def _checkProperties(self, input_shape, block_shape, base_paddings, paddings, argument
634 if base_paddings is None:
635 base_paddings = np.zeros((num_block_dims, 2), np.int32)
636 self.assertEqual(base_paddings.shape, (num_block_dims, 2))
640 self.assertEqual(paddings[i, 0], base_paddings[i, 0])
641 self.assertLessEqual(0, paddings[i, 1] - base_paddings[i, 1])
642 self.assertLess(paddings[i, 1] - base_paddings[i, 1], block_shape[i])
647 self.assertEqual(crops[i, 1], paddings[i, 1] - base_paddings[i, 1])
649 def _test(self, input_shape, block_shape, base_paddings): argument
652 if base_paddings is not None:
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 2605 base_paddings=None, argument
2650 if base_paddings is not None:
2651 base_paddings = ops.convert_to_tensor(
2652 base_paddings, dtype=dtypes.int32, name="base_paddings")
2653 base_paddings.get_shape().assert_is_compatible_with([num_block_dims, 2])
2655 base_paddings = zeros([num_block_dims, 2], dtypes.int32)
2659 const_base_paddings = tensor_util.constant_value(base_paddings)
2664 base_paddings = const_base_paddings
2667 pad_start = base_paddings[:, 0]
2668 orig_pad_end = base_paddings[:, 1]
|
| D | nn_ops.py | 568 self.base_paddings = _with_space_to_batch_base_paddings(
573 self.base_paddings = None
575 self.base_paddings = np.zeros([num_spatial_dims, 2], np.int32)
600 base_paddings = self.base_paddings
601 if base_paddings is None:
605 base_paddings = _with_space_to_batch_base_paddings(
609 base_paddings=base_paddings,
652 base_paddings = array_ops.stack(
654 return base_paddings
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 873 …argspec: "args=[\'input_shape\', \'block_shape\', \'base_paddings\', \'name\'], varargs=None, keyw…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1861 …argspec: "args=[\'input_shape\', \'block_shape\', \'base_paddings\', \'name\'], varargs=None, keyw…
|