Home
last modified time | relevance | path

Searched defs:out_num_axis (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dreduce_utils.h26 int& out_num_axis) { in RemoveSize1Dims()
59 int& out_num_axis, const int* shape_in, int* shape_out, in ResolveAxis()
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc96 int* out_axis, int* out_num_axis) { in ResolveAxis()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreduce.h111 int* out_num_axis) { in ResolveAxis()