| /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/core/ops/ |
| D | array_ops_test.cc | 1425 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1452 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1459 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1469 Tensor block_shape = test::AsTensor<int32>({3, 3}); in TEST() local
1479 Tensor block_shape = test::AsTensor<int32>({}); in TEST() local
1552 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1583 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1590 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1601 Tensor block_shape = test::AsTensor<int32>({2, 2}); in TEST() local
1612 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
|
| D | array_ops.cc | 2416 Tensor block_shape(tensorflow::DT_INT64, TensorShape({2})); in __anon42d741193402() local
2456 Tensor block_shape(tensorflow::DT_INT64, TensorShape({2})); in __anon42d741193602() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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 | space_to_batch_nd.cc | 45 const TfLiteTensor* block_shape; member
60 const int32* block_shape = GetTensorData<int32>(op_context->block_shape); in ResizeOutputTensor() local
|
| D | batch_to_space_nd_test.cc | 71 std::initializer_list<int> block_shape, in BatchToSpaceNDOpConstModel()
|
| D | space_to_batch_nd_test.cc | 76 const TensorData& input, std::initializer_list<int> block_shape, in SpaceToBatchNDOpConstModel()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | batch_to_space_nd.cc | 64 const TfLiteEvalTensor* block_shape = in Eval() local
|
| D | space_to_batch_nd.cc | 74 const TfLiteEvalTensor* block_shape = in Eval() 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 | 38 int32 block_shape[NUM_BLOCK_DIMS]; member
|
| D | spacetobatch_op.cc | 71 gtl::InlinedVector<int64, 4> block_shape; in SpaceToBatchOpCompute() local
|
| D | batchtospace_op.cc | 68 gtl::InlinedVector<int64, 4> block_shape; in BatchToSpaceOpCompute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3872 block_shape, argument
3957 block_shape=None): # pylint: disable=redefined-builtin argument
3975 def space_to_batch_v2(input, block_shape, paddings, name=None): # pylint: disable=redefined-builtin argument
4022 def batch_to_space(input, crops, block_size, name=None, block_shape=None): # pylint: disable=redef… argument
4040 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 | 192 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/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 873 ElementsAttr block_shape; in matchAndRewrite() local
1043 DenseIntElementsAttr block_shape; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 823 auto block_shape = block_shape_type.getShape(); in convertSpaceToBatchNDOp() local
1096 SmallVector<int64_t, 4> block_shape(block_rank); in convertBatchToSpaceNDOp() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2451 std::initializer_list<int> block_shape, in SpaceToBatchNDOpConstModel()
|