Home
last modified time | relevance | path

Searched refs:out_axis (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreduce.h59 const int64_t num_axis, int* out_axis, in ResolveAxis() argument
78 if (out_axis[j] == current) { in ResolveAxis()
84 out_axis[*out_num_axis] = current; in ResolveAxis()
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc96 int* out_axis, int* out_num_axis) { in ResolveAxis() argument
112 out_axis[*out_num_axis] = current; in ResolveAxis()