| /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;
197 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
198 op.input_tensors[1] = &size_tensor;
225 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
226 op.input_tensors[3] = &size_tensor;
251 Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
252 op.input_tensors[1] = &size_tensor;
310 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); local
311 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>();
|
| /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);
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| backports_test.cc | 113 Tensor size_tensor(DT_INT32, {});
114 size_tensor.flat<int32>()(0) = 1;
115 SetNodeTensorAttr<float>("value", size_tensor, size_node);
|
| quantize_nodes_test.cc | 659 Tensor size_tensor(DT_INT32, TensorShape({2}));
660 test::FillValues<int32>(&size_tensor, {256, 256});
663 Input::Initializer(size_tensor));
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| tensor_array_ops.cc | 553 Tensor size_tensor(DT_INT32, {});
554 size_tensor.scalar<int32>()() =
556 ctx->SetConstantOutput(0, size_tensor);
|
| /external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
| multi_head.py | 275 size_tensor = array_ops.concat(
278 logits, begin=begin_tensor, size=size_tensor)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 875 Tensor size_tensor; local
[all...] |
