Searched refs:num_output_tensors (Results 1 – 6 of 6) sorted by relevance
60 int num_output_tensors() const { in num_output_tensors() function61 int num_output_tensors = 0; in num_output_tensors() local64 num_output_tensors += output.length; in num_output_tensors()67 return num_output_tensors; in num_output_tensors()
436 int num_output_tensors = op_kernel.num_output_tensors(); in call_function() local438 tensor_id == num_tensors - num_output_tensors, in call_function()442 num_tensors - num_output_tensors); in call_function()
97 const unsigned int num_output_tensors = (num == 0) ? axis_size : std::min(axis_size, num); in compute_reference() local103 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()
174 int num_output_tensors = output_tensor_metadata->size(); in CheckAndSetOutputs() local176 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()
66 int num_output_tensors) { in GenerateTfLiteDelegateParams() argument70 auto* output_tensors = TfLiteIntArrayCreate(num_output_tensors); in GenerateTfLiteDelegateParams()80 for (int i = 0; i < num_output_tensors; ++i) { in GenerateTfLiteDelegateParams()
33 typedef void *(*tflitewrapper_create)(int num_input_tensors, int num_output_tensors);