Home
last modified time | relevance | path

Searched refs:base_paddings (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetobatch_op_test.py629 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/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc46 std::vector<int> base_paddings(4, 0); in ComputeSpaceToBatchParams() local
52 base_paddings[0] = (dilated_weights_h - 1) / 2; in ComputeSpaceToBatchParams()
53 base_paddings[1] = dilated_weights_h - 1 - (dilated_weights_h - 1) / 2; in ComputeSpaceToBatchParams()
54 base_paddings[2] = (dilated_weights_w - 1) / 2; in ComputeSpaceToBatchParams()
55 base_paddings[3] = dilated_weights_w - 1 - (dilated_weights_w - 1) / 2; in ComputeSpaceToBatchParams()
66 int pad_start_h = base_paddings[0]; in ComputeSpaceToBatchParams()
67 int pad_start_w = base_paddings[2]; in ComputeSpaceToBatchParams()
68 int orig_pad_end_h = base_paddings[1]; in ComputeSpaceToBatchParams()
69 int orig_pad_end_w = base_paddings[3]; in ComputeSpaceToBatchParams()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py667 self.base_paddings = _with_space_to_batch_base_paddings(
672 self.base_paddings = None
674 self.base_paddings = np.zeros([num_spatial_dims, 2], np.int32)
676 base_paddings = (np.array(explicit_paddings)
680 self.base_paddings = base_paddings[2:]
682 self.base_paddings = base_paddings[1:-1]
707 base_paddings = self.base_paddings
708 if base_paddings is None:
712 base_paddings = _with_space_to_batch_base_paddings(
717 base_paddings=base_paddings,
[all …]
Darray_ops.py3873 base_paddings=None, argument
3918 if base_paddings is not None:
3919 base_paddings = ops.convert_to_tensor(
3920 base_paddings, dtype=dtypes.int32, name="base_paddings")
3921 base_paddings.get_shape().assert_is_compatible_with([num_block_dims, 2])
3923 base_paddings = zeros([num_block_dims, 2], dtypes.int32)
3927 const_base_paddings = tensor_util.constant_value(base_paddings)
3932 base_paddings = const_base_paddings
3935 pad_start = base_paddings[:, 0]
3936 orig_pad_end = base_paddings[:, 1]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt945 …argspec: "args=[\'input_shape\', \'block_shape\', \'base_paddings\', \'name\'], varargs=None, keyw…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1945 …argspec: "args=[\'input_shape\', \'block_shape\', \'base_paddings\', \'name\'], varargs=None, keyw…