Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreduce.h60 int* out_num_axis) { in ResolveAxis() argument
61 *out_num_axis = 0; // Just in case. in ResolveAxis()
77 for (int j = 0; j < *out_num_axis; ++j) { in ResolveAxis()
84 out_axis[*out_num_axis] = current; in ResolveAxis()
85 *out_num_axis += 1; 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
97 *out_num_axis = 0; in ResolveAxis()
112 out_axis[*out_num_axis] = current; in ResolveAxis()
113 *out_num_axis += 1; in ResolveAxis()