ArgMax.cpp File Reference

#include "kernels/ArgMax.h"
#include "kernels/Utils.h"
#include "PALArgMax.h"

Go to the source code of this file.

Namespaces
namespace	luci_interpreter
namespace	luci_interpreter::kernels

Macros
#define	TF_LITE_ARG_MAX(data_type, axis_type, output_type)

Macro Definition Documentation

TF_LITE_ARG_MAX

#define TF_LITE_ARG_MAX	(		data_type,
			axis_type,
			output_type
	)

Value:

  luci_interpreter_pal::ArgMinMax(getTensorShape(input()), getTensorData<data_type>(input()), \
                                  getTensorData<axis_type>(axis()), getTensorShape(output()), \
                                  getTensorData<output_type>(output()), std::greater<data_type>())