Home
last modified time | relevance | path

Searched defs:output_tensor (Results 1 – 25 of 165) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1550 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAbsNode() local
1590 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAddNode() local
1634 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAveragePool2DNode() local
1699 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitCeilNode() local
1767 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitConv2DNode() local
1871 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDepthwiseConv2DNode() local
1951 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDepthToSpaceNode() local
2003 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDivNode() local
2046 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitEluNode() local
2112 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitFullyConnectedNode() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dbase64_ops.cc35 Tensor* output_tensor = nullptr; in Compute() local
60 Tensor* output_tensor = nullptr; in Compute() local
Ddecode_raw_op.cc62 Tensor* output_tensor = nullptr; in Compute() local
74 Tensor* output_tensor = nullptr; in Compute() local
Dregex_full_match_op.cc51 Tensor* output_tensor = nullptr; in Compute() local
104 Tensor* output_tensor = nullptr; in Compute() local
Ddebug_ops.h164 Tensor* output_tensor; in ApplyGrpcGating() local
235 Tensor* output_tensor; in Compute() local
278 Tensor* output_tensor; in Compute() local
Dstring_to_number_op.cc45 Tensor* output_tensor = nullptr; in Compute() local
Dstring_strip_op.cc36 Tensor* output_tensor; in Compute() local
Dstring_to_hash_bucket_fast_op.h41 Tensor* output_tensor = nullptr; in Compute() local
Dunicode_script_op.cc31 Tensor* output_tensor = nullptr; in Compute() local
Dstring_to_hash_bucket_op.h48 Tensor* output_tensor = nullptr; in Compute() local
Dstring_to_hash_bucket_op.cc37 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/kernels/
Dhashtable_size.cc41 TfLiteTensor* output_tensor; in PrepareHashtableSize() local
56 TfLiteTensor* output_tensor; in EvalHashtableSize() local
Dhashtable_find.cc50 TfLiteTensor* output_tensor; in PrepareHashtableFind() local
74 TfLiteTensor* output_tensor; in EvalHashtableFind() local
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_segmenter.cc216 const TfLiteTensor* output_tensor = TfLiteEngine::GetOutput(interpreter, 0); in CheckAndSetOutputs() local
323 const TfLiteTensor* output_tensor = output_tensors[0]; in Postprocess() local
412 float ImageSegmenter::GetOutputConfidence(const TfLiteTensor& output_tensor, in GetOutputConfidence()
Dimage_classifier.cc187 const tflite::TensorMetadata* output_tensor = in CheckAndSetOutputs() local
220 const TfLiteTensor* output_tensor = in CheckAndSetOutputs() local
424 const TfLiteTensor* output_tensor = output_tensors[i]; in Postprocess() local
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg.cc64 TfLiteTensor* output_tensor; in Prepare() local
98 TfLiteTensor* output_tensor; in Eval() local
/external/tensorflow/tensorflow/core/user_ops/
Dfact.cc33 tensorflow::Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dackermann_op.cc35 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/examples/adding_an_op/
Dcuda_op_kernel.cc43 Tensor* output_tensor = nullptr; in Compute() local
Dzero_out_op_kernel_3.cc53 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/
Dcuda_op_kernel.cc43 Tensor* output_tensor = NULL; in Compute() local
Dzero_out_op_kernel_1.cc46 Tensor* output_tensor = NULL; in Compute() local
/external/tensorflow/tensorflow/c/
Dc_test_util.h149 TF_Tensor* output_tensor(int i) { return output_values_[i]; } in output_tensor() function
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dsoftmax_op_builder.cc53 TensorID output_tensor = GetOutput(context); in RegisterOutputs() local
/external/tensorflow/tensorflow/lite/python/testdata/
Ddouble_op.cc41 Tensor* output_tensor = nullptr; in Compute() local

1234567