| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops_test.cc | 48 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
49 op.input_tensors[1] = &size_tensor; in TEST()
233 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
234 op.input_tensors[1] = &size_tensor; in TEST()
261 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
262 op.input_tensors[3] = &size_tensor; in TEST()
287 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
288 op.input_tensors[1] = &size_tensor; in TEST()
346 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
347 op.input_tensors.at(1) = &size_tensor; in TEST()
|
| D | math_ops.cc | 1652 const Tensor* size_tensor = c->input_tensor(1); in __anon725c8f640e02() local
1653 if (size_tensor == nullptr) { in __anon725c8f640e02()
1660 int32 size_val = size_tensor->scalar<int32>()(); in __anon725c8f640e02()
1684 const Tensor* size_tensor = c->input_tensor(1); in __anon725c8f640f02() local
1685 if (size_tensor == nullptr) { in __anon725c8f640f02()
1695 size_val = static_cast<int64>(size_tensor->scalar<int32>()()); in __anon725c8f640f02()
1697 size_val = size_tensor->scalar<int64>()(); in __anon725c8f640f02()
1725 const Tensor* size_tensor = c->input_tensor(3); in __anon725c8f641002() local
1726 if (size_tensor == nullptr) { in __anon725c8f641002()
1736 size_val = static_cast<int64>(size_tensor->scalar<int32>()()); in __anon725c8f641002()
[all …]
|
| D | image_ops.cc | 39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
42 if (size_tensor == nullptr) { in SetOutputToSizedImage()
47 if (size_tensor->dtype() != DT_INT32) { in SetOutputToSizedImage()
51 DataTypeString(size_tensor->dtype()), " for input #", size_input_idx, in SetOutputToSizedImage()
54 auto vec = size_tensor->vec<int32>(); in SetOutputToSizedImage()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | resize_bilinear_builder.cc | 35 const auto& size_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
36 if (!IsConstantTensor(&size_tensor)) { in PopulateSubGraph()
43 std::vector<int> dims = {size_tensor.data.i32[0], size_tensor.data.i32[1]}; in PopulateSubGraph()
|
| D | slice_builder.cc | 45 const auto& size_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
49 &size_tensor, &begins, &sizes); in PopulateSubGraph()
52 &size_tensor, &begins, &sizes); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 69 const Tensor& size_tensor = context->input(2); in SharedSliceValidation() local
74 TensorShapeUtils::IsVector(size_tensor.shape()) && in SharedSliceValidation()
76 size_tensor.NumElements() == input.dims(), in SharedSliceValidation()
80 " and ", size_tensor.shape().DebugString(), " instead.")); in SharedSliceValidation()
84 IntTensorToInt64Vec(size_tensor, size); in SharedSliceValidation()
|
| D | bincount_op.cc | 177 const Tensor& size_tensor = ctx->input(1); in Compute() local
178 OP_REQUIRES(ctx, size_tensor.dims() == 0, in Compute()
180 size_tensor.dims())); in Compute()
181 int32 size = size_tensor.scalar<int32>()(); in Compute()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | backports_test.cc | 114 Tensor size_tensor(DT_INT32, {}); in TEST() local
115 size_tensor.flat<int32>()(0) = 1; in TEST()
116 SetNodeTensorAttr<float>("value", size_tensor, size_node); in TEST()
|
| D | quantize_nodes_test.cc | 659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
660 test::FillValues<int32>(&size_tensor, {256, 256}); in TestQuantizeResizeBilinear()
663 Input::Initializer(size_tensor)); in TestQuantizeResizeBilinear()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.cc | 372 const auto& size_tensor = context->tensors[node->inputs->data[1]]; in IsNodeSupportedByHexagon() local
375 return NumElements(&size_tensor) == 2 && size_tensor.data.i32[0] < 66 && in IsNodeSupportedByHexagon()
376 size_tensor.data.i32[1] < 66; in IsNodeSupportedByHexagon()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 71 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
90 TensorShapeUtils::IsVector(size_tensor.shape()) && in ValidateMklInputs()
92 size_tensor.NumElements() == input_dims, in ValidateMklInputs()
96 " and ", size_tensor.shape().DebugString(), " instead.")); in ValidateMklInputs()
99 *size = IntTensorToInt64Vec(size_tensor); in ValidateMklInputs()
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 281 auto* size_tensor = interpreter_->tensor(kSizeTensorId); in GetTableSize() local
282 auto size_tensor_shape = GetTensorShape(size_tensor); in GetTableSize()
283 return GetTensorData<int64_t>(size_tensor)[0]; in GetTableSize()
287 auto* size_tensor = interpreter_->tensor(kSizeTwoTensorId); in GetTableTwoSize() local
288 auto size_tensor_shape = GetTensorShape(size_tensor); in GetTableTwoSize()
289 return GetTensorData<int64_t>(size_tensor)[0]; in GetTableTwoSize()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 560 Tensor size_tensor(DT_INT32, {}); in Compile() local
561 size_tensor.scalar<int32>()() = static_cast<int32>(var->max_array_size()); in Compile()
562 ctx->SetConstantOutput(0, size_tensor); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 183 Value size_tensor = in matchAndRewrite() local
187 lhs_is_scalar ? rhs : lhs, size_tensor); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 980 Tensor size_tensor; in TEST_F() local
981 EXPECT_TRUE(size_tensor.FromProto( in TEST_F()
985 test::ExpectTensorEqual<int>(size_tensor_expected, size_tensor); in TEST_F()
|