Home
last modified time | relevance | path

Searched defs:size_tensor (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dimage_ops_test.cc48 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
233 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
261 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
287 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
346 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
Dmath_ops.cc1652 const Tensor* size_tensor = c->input_tensor(1); in __anon725c8f640e02() local
1684 const Tensor* size_tensor = c->input_tensor(1); in __anon725c8f640f02() local
1725 const Tensor* size_tensor = c->input_tensor(3); in __anon725c8f641002() local
Dimage_ops.cc39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dresize_bilinear_builder.cc35 const auto& size_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
Dslice_builder.cc45 const auto& size_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports_test.cc114 Tensor size_tensor(DT_INT32, {}); in TEST() local
Dquantize_nodes_test.cc659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc372 const auto& size_tensor = context->tensors[node->inputs->data[1]]; in IsNodeSupportedByHexagon() local
/external/tensorflow/tensorflow/core/kernels/
Dslice_op.cc69 const Tensor& size_tensor = context->input(2); in SharedSliceValidation() local
Dbincount_op.cc177 const Tensor& size_tensor = ctx->input(1); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_ops_test.cc281 auto* size_tensor = interpreter_->tensor(kSizeTensorId); in GetTableSize() local
287 auto* size_tensor = interpreter_->tensor(kSizeTwoTensorId); in GetTableTwoSize() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc71 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_array_ops.cc560 Tensor size_tensor(DT_INT32, {}); in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtransform_unranked_hlo.cc183 Value size_tensor = in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc980 Tensor size_tensor; in TEST_F() local