Searched defs:crops (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 27 const xla::Literal& crops) { in BatchToSpace()
154 xla::Literal crops; in Compile() local
176 xla::Literal crops; in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd.cc | 45 const TfLiteTensor* crops; member
60 const int* crops = GetTensorData<int32>(op_context->crops); in ResizeOutputTensor() local
|
| D | batch_to_space_nd_test.cc | 65 std::initializer_list<int> crops, in BatchToSpaceNDOpConstModel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batchtospace_op_test.py | 251 def _checkGrad(self, x, crops, block_size): argument
310 def _checkGrad(self, x, block_shape, crops, crops_dtype): argument
328 def _compare(self, input_shape, block_shape, crops, crops_dtype): argument
|
| D | spacetobatch_op_test.py | 630 crops): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batchtospace_op.cc | 69 gtl::InlinedVector<int64, 8> crops; in BatchToSpaceOpCompute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1503 Tensor crops = test::AsTensor<int32>({1, 1, 0, 1}, {{2, 2}}); in TEST() local
1511 Tensor crops = test::AsTensor<int32>({1, 1, 0, 0}, {{2, 2}}); in TEST() local
1546 Tensor crops = test::AsTensor<int32>({3, 2, 0, 0}, {{2, 2}}); in TEST() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 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/compiler/tests/ |
| D | randomized_tests.cc | 1250 Tensor crops; in TEST_F() local
1288 Tensor crops; in TEST_F() local
|