Home
last modified time | relevance | path

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

123456789

/external/tensorflow/tensorflow/examples/custom_ops_doc/sleep/
Dsleep_kernel.cc69 Tensor* output_tensor = nullptr; in ComputeAsync() local
86 void sleeper(Tensor* output_tensor, absl::Time when, DoneCallback done) { in sleeper()
122 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1768 const TfLiteTensor& output_tensor, int dimension_index, int node_index, in CheckTensorsDimensionMatch()
1812 const TfLiteTensor& output_tensor, float scale_min, float scale_max, in CheckTensorsInputOutputScale()
1840 const TfLiteTensor& input2_tensor, const TfLiteTensor& output_tensor, in CheckTensorsInputProductOutputScale()
2159 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAbsNode() local
2201 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAddNode() local
2255 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitAveragePool2DNode() local
2321 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitCeilNode() local
2386 const TfLiteTensor& output_tensor = tensors[output_idx]; in VisitSplitNode() local
2464 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitConcatenationNode() local
2596 const TfLiteTensor& output_tensor = tensors[node->outputs->data[0]]; in VisitConv2DNode() 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.h165 Tensor* output_tensor; in ApplyGrpcGating() local
236 Tensor* output_tensor; in Compute() local
279 Tensor* output_tensor; in Compute() local
Dstring_strip_op.cc36 Tensor* output_tensor; in Compute() local
Dstring_to_number_op.cc45 Tensor* output_tensor = nullptr; in Compute() local
Dunicode_script_op.cc31 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/c/
Dc_test.c121 const TfLiteTensor* output_tensor = in TestInferenceUsingSignature() local
179 const TfLiteTensor* output_tensor = in TestRepeatResizeInputTensor() local
277 const TfLiteTensor* output_tensor = in TestInferenceUsingInterpreter() local
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg.cc81 TfLiteTensor* output_tensor; in Prepare() local
117 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/custom_ops/
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
/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/federated-compute/fcp/tensorflow/
Dtensor_name_op.cc52 tensorflow::Tensor* output_tensor; in Compute() local
Dcrc32_op.cc52 Tensor* output_tensor = nullptr; in Compute() local
Dexample_selector_fuser_op.cc89 Tensor* output_tensor = nullptr; in Compute() local
/external/mesa3d/src/gallium/targets/teflon/
Dtest_executor.cpp153 auto output_tensor = subgraph->tensors[output_index]; in patch_conv2d() local
236 auto output_tensor = subgraph->tensors[output_index]; in add_generate_model() local
310 auto output_tensor = subgraph->tensors[output_index]; in patch_fully_connected() local
480 const TfLiteTensor *output_tensor = TfLiteInterpreterGetOutputTensor(interpreter, i); in run_model() local
/external/tensorflow/tensorflow/c/
Dc_test_util.h149 TF_Tensor* output_tensor(int i) { return output_values_[i]; } in output_tensor() function

123456789