Home
last modified time | relevance | path

Searched defs:crops (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc27 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/
Dbatch_to_space_nd.cc45 const TfLiteTensor* crops; member
60 const int* crops = GetTensorData<int32>(op_context->crops); in ResizeOutputTensor() local
Dbatch_to_space_nd_test.cc72 std::initializer_list<int> crops, in BatchToSpaceNDOpConstModel()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbatchtospace_op_test.py251 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
Dspacetobatch_op_test.py630 crops): argument
/external/tensorflow/tensorflow/lite/micro/kernels/
Dbatch_to_space_nd.cc66 const TfLiteEvalTensor* crops = in Eval() local
Dspace_to_batch_nd.cc76 const TfLiteEvalTensor* crops = in Eval() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddilated_conv.h245 auto crops = bts_crops_attr.getValues<IntegerAttr>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc69 gtl::InlinedVector<int64, 8> crops; in BatchToSpaceOpCompute() local
/external/minigbm/cros_gralloc/gralloc4/
DCrosGralloc4Mapper.cc558 std::vector<aidl::android::hardware::graphics::common::Rect> crops; in get() local
716 std::vector<aidl::android::hardware::graphics::common::Rect> crops; in getFromBufferDescriptorInfo() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc1560 Tensor crops = test::AsTensor<int32>({1, 1, 0, 1}, {{2, 2}}); in TEST() local
1568 Tensor crops = test::AsTensor<int32>({1, 1, 0, 0}, {{2, 2}}); in TEST() local
1603 Tensor crops = test::AsTensor<int32>({3, 2, 0, 0}, {{2, 2}}); in TEST() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc43 std::vector<int>* crops) { in ComputeSpaceToBatchParams()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py4022 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/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1044 DenseIntElementsAttr crops; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1251 Tensor crops; in TEST_F() local
1289 Tensor crops; in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1097 SmallVector<std::pair<int64_t, int64_t>, 4> crops(crops_dims); in convertBatchToSpaceNDOp() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc2207 auto crops = context->tensors[node->inputs->data[2]]; in Validate() local