| /external/tensorflow/tensorflow/core/ops/ |
| image_ops_test.cc | 47 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
48 op.input_tensors[1] = &size_tensor;
204 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
205 op.input_tensors[1] = &size_tensor;
232 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
233 op.input_tensors[3] = &size_tensor;
258 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
259 op.input_tensors[1] = &size_tensor;
317 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
318 op.input_tensors.at(1) = &size_tensor;
[all...] |
| image_ops.cc | 29 // height and width come from the size_tensor.
39 const Tensor* size_tensor = c->input_tensor(size_input_idx); local
42 if (size_tensor == nullptr) {
47 if (size_tensor->dtype() != DT_INT32) {
51 DataTypeString(size_tensor->dtype()), " for input #", size_input_idx,
54 auto vec = size_tensor->vec<int32>();
[all...] |
| /external/tensorflow/tensorflow/contrib/image/ops/ |
| image_ops.cc | 29 // height and width come from the size_tensor.
39 const Tensor* size_tensor = c->input_tensor(size_input_idx); local
42 if (size_tensor == nullptr) {
47 if (size_tensor->dtype() != DT_INT32) {
51 DataTypeString(size_tensor->dtype()), " for input #", size_input_idx,
54 auto vec = size_tensor->vec<int32>();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| bincount_op.cc | 96 const Tensor& size_tensor = ctx->input(1); variable
99 int32 size = size_tensor.scalar<int32>()();
|
| slice_op.cc | 72 const Tensor& size_tensor = context->input(2); local
77 context->op_kernel().IsLegacyVector(size_tensor.shape()) &&
79 size_tensor.NumElements() == input.dims(),
83 " and ", size_tensor.shape().DebugString(), " instead."));
87 *size = IntTensorToInt64Vec(size_tensor);
|
| mkl_slice_op.cc | 74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); local
92 context->op_kernel().IsLegacyVector(size_tensor.shape()) &&
94 size_tensor.NumElements() == input_dims,
98 " and ", size_tensor.shape().DebugString(), " instead."));
101 *size = IntTensorToInt64Vec(size_tensor);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 960 Tensor size_tensor; local
[all...] |
