Searched refs:kAxis (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | expand_dims.cc | 30 enum { kInput = 0, kAxis }; enumerator
78 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Prepare()
99 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Eval()
|
| D | arg_min_max.cc | 35 constexpr int kAxis = 1; variable
70 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Prepare()
135 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | arg_min_max.cc | 31 constexpr int kAxis = 1; variable
52 tflite::micro::GetEvalInput(context, node, kAxis); in Eval()
|