| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batchtospace_op.cc | 27 const xla::Literal& crops) {
42 xla::ShapeUtil::Rank(crops.shape()) == 2 &&
43 block_rank == xla::ShapeUtil::GetDimension(crops.shape(), 0) &&
44 2 == xla::ShapeUtil::GetDimension(crops.shape(), 1),
45 errors::InvalidArgument("crops should have shape [", block_rank,
47 xla::ShapeUtil::HumanString(crops.shape())));
118 // `reshaped_permuted` according to `crops` to produce the output of shape:
121 // input_shape[1] * block_shape[0] - crops[0,0] - crops[0,1],
123 // input_shape[M] * block_shape[M-1] - crops[M-1,0] - crops[M-1,1]
155 xla::Literal crops; variable
175 xla::Literal crops; variable
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| batch_to_space_nd.cc | 40 crops = GetInput(context, node, 2);
45 TfLiteTensor* crops; member in struct:tflite::ops::builtin::batch_to_space_nd::BatchToSpaceNDContext
60 const int* crops = GetTensorData<int32>(op_context->crops); local
66 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->crops),
69 // TODO(ycling): Add crops as part of calculation. Remove check for a crops
71 TF_LITE_ENSURE_EQ(context, crops[0], 0);
72 TF_LITE_ENSURE_EQ(context, crops[1], 0);
73 TF_LITE_ENSURE_EQ(context, crops[2], 0)
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops_test.cc | 1452 Tensor crops = test::AsTensor<int32>({1, 1, 0, 1}, {{2, 2}}); local
1460 Tensor crops = test::AsTensor<int32>({1, 1, 0, 0}, {{2, 2}}); local
1495 Tensor crops = test::AsTensor<int32>({3, 2, 0, 0}, {{2, 2}}); local
[all...] |
| /frameworks/native/libs/gui/tests/ |
| SurfaceTextureGL_test.cpp | 140 android_native_rect_t crops[] = { local
149 const android_native_rect_t& crop(crops[i]);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| batchtospace_op.cc | 63 errors::InvalidArgument("crops should have shape [", block_dims,
67 // crops tensors are concurrently modified, we must copy the values.
69 gtl::InlinedVector<int64, 8> crops; local
71 internal::spacetobatch::SubtleMustCopyFlat(orig_crops, &crops);
78 if (crops[2 * dim] != 0 || crops[2 * dim + 1] != 0 ||
90 if (crops[2 * dim] != 0 || crops[2 * dim + 1] != 0 ||
150 const int64 crop_start = crops[2 * block_dim],
151 crop_end = crops[2 * block_dim + 1]
[all...] |
