Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1053 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAbsNode() local
1093 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAddNode() local
1137 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAveragePool2DNode() local
1202 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitCeilNode() local
1269 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitConv2DNode() local
1364 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDepthwiseConv2DNode() local
1435 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDepthToSpaceNode() local
1487 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitDivNode() local
1530 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitEluNode() local
1597 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.cc60 Tensor* output_tensor = nullptr; in Compute() local
72 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
Dunicode_script_op.cc31 Tensor* output_tensor = nullptr; in Compute() local
Dstring_to_hash_bucket_fast_op.h41 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_size.cc43 TfLiteTensor* output_tensor; in PrepareHashtableSize() local
58 TfLiteTensor* output_tensor; in EvalHashtableSize() local
Dhashtable_find.cc53 TfLiteTensor* output_tensor; in PrepareHashtableFind() local
77 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/micro/kernels/
Dreshape_test.cc53 TfLiteTensor* output_tensor = &tensors[outputs_array->data[0]]; in ValidateReshapeGoldens() local
67 TfLiteTensor* output_tensor, const T* expected_output, in TestReshapeWithShape()
92 TfLiteTensor* output_tensor, in TestReshapeWithoutShape()
126 TfLiteTensor output_tensor = CreateTensor(output_data, output_dims); in TestReshape() local
148 TfLiteTensor output_tensor = CreateQuantizedTensor( in TestReshapeQuantized() local
221 auto output_tensor = tflite::testing::CreateTensor(output_data, output_dims); in TF_LITE_MICRO_TEST() local
340 auto output_tensor = CreateTensor(output_data, output_dims); in TF_LITE_MICRO_TEST() local
Dconcatenation.cc40 inline int CalculatePositiveAxis(int axis, const TfLiteTensor* output_tensor) { in CalculatePositiveAxis()
142 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Prepare() local
230 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Eval() local
Dsvdf_common.cc39 TfLiteEvalTensor* output_tensor, in EvalIntegerSvdfReference()
171 tflite::micro::GetTensorData<int8_t>(output_tensor)[i] = in EvalIntegerSvdfReference() 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/python/testdata/
Ddouble_op.cc39 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dsoftmax_op_builder.cc53 TensorID output_tensor = GetOutput(context); in RegisterOutputs() local

1234567