Home
last modified time | relevance | path

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

123456789

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/sleep/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc1414 ITensorProxyPtr* output_tensor, in TransposeTensor()
2021 ITensorProxyPtr output_tensor = conv_layer->getOutput(0); in ConvertConv2DHelper() local
2119 ITensorProxyPtr output_tensor = nullptr; in ConvertTranspose() local
2206 ITensorProxyPtr* output_tensor) { in ConvertStaticReshapeForExplicitBatchMode()
2215 ITensorProxyPtr* output_tensor) { in ConvertStaticReshapeForImplicitBatchMode()
2272 ITensorProxyPtr output_tensor = nullptr; in ConvertReshape() local
2310 ITensorProxyPtr output_tensor = nullptr; in ConvertExpandDims() local
2507 ITensorProxyPtr output_tensor = nullptr; in ConvertSqueeze() local
2967 ITensorProxyPtr output_tensor = conv_layer->getOutput(0); in ConvertConv3DHelper() local
3063 ITensorProxyPtr output_tensor = layer->getOutput(0); in ConvertPool3D() local
[all …]
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/vision/
H A 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()
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A 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
H A Dc_api_test.cc84 const TfLiteTensor* output_tensor = in TEST() local
142 const TfLiteTensor* output_tensor = in TEST() local
348 const TfLiteTensor* output_tensor = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_builder.cc130 const TfLitePadding padding_type, TensorID* output_tensor, in BuildDilatedDwConv()
244 const TfLitePadding padding_type, TensorID* output_tensor, in BuildSplittedDwConv()
333 TensorID* output_tensor, TensorID* output_min_tensor, in BuildStandardConv()
477 TensorID output_tensor, output_min_tensor, output_max_tensor; in PopulateSubGraph() local
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp146 auto output_tensor = subgraph->tensors[output_index]; in patch_conv2d() local
215 auto output_tensor = subgraph->tensors[output_index]; in add_generate_model() local
303 const TfLiteTensor *output_tensor = TfLiteInterpreterGetOutputTensor(interpreter, i); in run_model() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddebug_ops.h165 Tensor* output_tensor; in ApplyGrpcGating() local
236 Tensor* output_tensor; in Compute() local
279 Tensor* output_tensor; in Compute() local
H A Dbase64_ops.cc35 Tensor* output_tensor = nullptr; in Compute() local
60 Tensor* output_tensor = nullptr; in Compute() local
H A Ddecode_raw_op.cc62 Tensor* output_tensor = nullptr; in Compute() local
74 Tensor* output_tensor = nullptr; in Compute() local
H A Dregex_full_match_op.cc51 Tensor* output_tensor = nullptr; in Compute() local
104 Tensor* output_tensor = nullptr; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc154 Tensor* output_tensor = nullptr; in Compute() local
194 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor()
411 Tensor* output_tensor = nullptr; in Compute() local
457 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor()
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py656 def replace(self, input_tensor, output_tensor) -> None: argument
704 def replace(self, input_tensor, output_tensor) -> None: argument
745 def replace(self, input_tensor, output_tensor) -> None: argument
791 def replace(self, input_tensor, output_tensor) -> None: argument
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc454 TfLiteTensor* output_tensor) { in SetOutputT()
529 TfLiteTensor* output_tensor) { in SetOutput()
593 TfLiteTensor& output_tensor = in Prepare() local
645 TfLiteTensor& output_tensor = in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dhashtable_size.cc41 TfLiteTensor* output_tensor; in PrepareHashtableSize() local
56 TfLiteTensor* output_tensor; in EvalHashtableSize() local
H A Dhashtable_find.cc50 TfLiteTensor* output_tensor; in PrepareHashtableFind() local
74 TfLiteTensor* output_tensor; in EvalHashtableFind() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_util.cc355 Tensor* output_tensor; in CreateExecuteOutput() local
368 Tensor* output_tensor; in CreateExecuteOutput() local
433 Tensor* output_tensor; in ExecuteChained() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Ddecode_jpeg.cc81 TfLiteTensor* output_tensor; in Prepare() local
117 TfLiteTensor* output_tensor; in Eval() local
/aosp_15_r20/test/mlts/benchmark/jni/
Drun_tflite.cpp277 auto* output_tensor = mTfliteInterpreter->tensor(output); in saveInferenceOutput() local
286 auto* output_tensor = mTfliteInterpreter->tensor(output); in getOutputError() local
741 auto* output_tensor = mTfliteInterpreter->tensor(tensor_index); in dumpAllLayers() local
/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp439 auto output_tensor = work->result(); in testSparseAllreduce() local
490 auto output_tensor = work->result(); in testSparseAllreduceLarge() local
617 auto output_tensor = test.getOutputTensor(); in testAllgatherBase() local
638 auto output_tensor = test.getOutputTensor(); in testReduceScatterBase() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature.cc168 const Tensor* output_tensor = in GetOpSignature() local
232 const Tensor* output_tensor = in GetOpSignature() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_ml_tp.c615 unsigned *output_tensor) in etna_ml_lower_transpose()
669 unsigned *output_tensor) in etna_ml_lower_reshuffle()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_execute_op.cc457 auto transfer_buffers = [&](int i, Tensor* output_tensor) { in AllocateOutputTensors()
527 Tensor* output_tensor; in AllocateOutputTensors() local
566 Tensor* output_tensor; in AllocateOutputTensors() local

123456789