Home
last modified time | relevance | path

Searched defs:num_output_tensors (Results 1 – 5 of 5) 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
H A Doss_proxy_executor.cpp436 int num_output_tensors = op_kernel.num_output_tensors(); in call_function() local
/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
/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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dserialization_test.cc66 int num_output_tensors) { in GenerateTfLiteDelegateParams()