Home
last modified time | relevance | path

Searched defs:ResolveAxis (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dreduce_utils.h57 inline bool ResolveAxis(const int num_dims, const int* axis, in ResolveAxis() function
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc95 inline bool ResolveAxis(const int num_dims, const int* axis, const int num_axis, in ResolveAxis() function
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreduce.h109 inline bool ResolveAxis(const int num_dims, const int* axis, in ResolveAxis() function
/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc365 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis() function