Searched refs:TF_LITE_ARG_MIN_MAX (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | arg_min_max.cc | 139 #define TF_LITE_ARG_MIN_MAX(data_type, axis_type, output_type) \ in Eval() macro
149 TF_LITE_ARG_MIN_MAX(float, int32_t, int32_t); in Eval()
152 TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int32_t); in Eval()
155 TF_LITE_ARG_MIN_MAX(int8_t, int32_t, int32_t); in Eval()
158 TF_LITE_ARG_MIN_MAX(int32_t, int32_t, int32_t); in Eval()
161 TF_LITE_ARG_MIN_MAX(bool, int32_t, int32_t); in Eval()
174 TF_LITE_ARG_MIN_MAX(float, int32_t, int64_t); in Eval()
177 TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int64_t); in Eval()
180 TF_LITE_ARG_MIN_MAX(int8_t, int32_t, int64_t); in Eval()
183 TF_LITE_ARG_MIN_MAX(int32_t, int32_t, int64_t); in Eval()
[all …]
|