| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | spacetobatch_op_test.py | 34 def space_to_batch_direct(input_array, block_shape, paddings): argument
200 def _testPad(self, inputs, block_shape, paddings, outputs): argument
214 def _testDirect(self, input_shape, block_shape, paddings): argument
422 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
431 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
449 def _testShape(self, input_shape, block_shape, paddings, error): argument
586 def _checkGrad(self, x, block_shape, paddings): argument
603 def _compare(self, input_shape, block_shape, paddings): argument
629 def _checkProperties(self, input_shape, block_shape, base_paddings, paddings, argument
649 def _test(self, input_shape, block_shape, base_paddings): argument
|
| D | batchtospace_op_test.py | 139 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
148 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
167 def _testShape(self, input_shape, block_shape, paddings, error): argument
310 def _checkGrad(self, x, block_shape, crops, crops_dtype): argument
328 def _compare(self, input_shape, block_shape, crops, crops_dtype): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | spacetobatch_op_test.py | 30 def space_to_batch_direct(input_array, block_shape, paddings): argument
155 def _testPad(self, inputs, block_shape, paddings, outputs): argument
187 def _testDirect(self, input_shape, block_shape, paddings): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 26 absl::Span<const int64> block_shape, in BatchToSpace()
151 std::vector<int64> block_shape; in Compile() local
|
| D | spacetobatch_op.cc | 26 absl::Span<const int64> block_shape, in SpaceToBatch()
153 std::vector<int64> block_shape; in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | space_to_batch_nd.cc | 44 const TfLiteTensor* block_shape; member
59 const int32* block_shape = GetTensorData<int32>(op_context->block_shape); in ResizeOutputTensor() local
|
| D | batch_to_space_nd.cc | 44 const TfLiteTensor* block_shape; member
59 const int* block_shape = GetTensorData<int32>(op_context->block_shape); in ResizeOutputTensor() local
|
| D | batch_to_space_nd_test.cc | 64 std::initializer_list<int> block_shape, in BatchToSpaceNDOpConstModel()
|
| D | space_to_batch_nd_test.cc | 72 std::initializer_list<int> block_shape, in SpaceToBatchNDOpConstModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1378 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1405 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1412 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1422 Tensor block_shape = test::AsTensor<int32>({3, 3}); in TEST() local
1495 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1526 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1533 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1544 Tensor block_shape = test::AsTensor<int32>({2, 2}); in TEST() local
1555 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
|
| D | array_ops.cc | 2259 Tensor block_shape(tensorflow::DT_INT64, TensorShape({2})); in __anon7c94107b3402() local
2299 Tensor block_shape(tensorflow::DT_INT64, TensorShape({2})); in __anon7c94107b3602() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | flatten_atrous.cc | 62 Tensor block_shape = in FlattenAtrousConv() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_functor.cc | 104 int64 block_shape[NUM_BLOCK_DIMS]; in operator ()() local
|
| D | spacetobatch_functor_gpu.cu.cc | 39 int32 block_shape[NUM_BLOCK_DIMS]; member
|
| D | batchtospace_op.cc | 68 gtl::InlinedVector<int64, 4> block_shape; in BatchToSpaceOpCompute() local
|
| D | spacetobatch_op.cc | 71 gtl::InlinedVector<int64, 4> block_shape; in SpaceToBatchOpCompute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 2604 block_shape, argument
2684 …input, paddings, block_size=None, name=None, block_shape=None): # pylint: disable=redefined-built… argument
2700 def space_to_batch_v2(input, block_shape, paddings, name=None): # pylint: disable=redefined-builtin argument
2742 def batch_to_space(input, crops, block_size, name=None, block_shape=None): # pylint: disable=redef… argument
2758 def batch_to_space_v2(input, block_shape, crops, name=None): # pylint: disable=redefined-builtin argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 193 def block_shape(self): member in _BaseLinearOperatorCirculant
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 287 auto block_shape = Const(scope_, {2, 2}); in TEST_F() local
306 auto block_shape = Const(scope_, {2, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1620 std::initializer_list<int> block_shape, in SpaceToBatchNDOpConstModel()
|