Home
last modified time | relevance | path

Searched refs:num_output_tensors (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Doss_proxy_executor.h60 int num_output_tensors() const { in num_output_tensors() function
61 int num_output_tensors = 0; in num_output_tensors() local
64 num_output_tensors += output.length; in num_output_tensors()
67 return num_output_tensors; in num_output_tensors()
H A Doss_proxy_executor.cpp436 int num_output_tensors = op_kernel.num_output_tensors(); in call_function() local
438 tensor_id == num_tensors - num_output_tensors, in call_function()
442 num_tensors - num_output_tensors); in call_function()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DUnstackFixture.h97 const unsigned int num_output_tensors = (num == 0) ? axis_size : std::min(axis_size, num); in compute_reference() local
103 std::vector<SimpleTensor<T>> output_tensors(num_output_tensors); in compute_reference()
104 for(size_t k = 0; k < num_output_tensors; ++k) in compute_reference()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/vision/
H A Dimage_classifier.cc174 int num_output_tensors = output_tensor_metadata->size(); in CheckAndSetOutputs() local
176 if (num_outputs_ != num_output_tensors) { in CheckAndSetOutputs()
182 num_outputs_, num_output_tensors), in CheckAndSetOutputs()
186 for (int i = 0; i < num_output_tensors; ++i) { in CheckAndSetOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dserialization_test.cc66 int num_output_tensors) { in GenerateTfLiteDelegateParams() argument
70 auto* output_tensors = TfLiteIntArrayCreate(num_output_tensors); in GenerateTfLiteDelegateParams()
80 for (int i = 0; i < num_output_tensors; ++i) { in GenerateTfLiteDelegateParams()
/aosp_15_r20/hardware/google/pixel/thermal/virtualtemp_estimator/
Dvirtualtemp_estimator_data.h33 typedef void *(*tflitewrapper_create)(int num_input_tensors, int num_output_tensors);