Searched refs:num_resolved_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 147 int num_resolved_axis = 0; in ReduceGeneric() local
149 &num_resolved_axis)) { in ReduceGeneric()
154 output_num_dims, resolved_axis, num_resolved_axis, in ReduceGeneric()
184 int num_resolved_axis = 0; in Mean() local
186 &num_resolved_axis)) { in Mean()
191 output_num_dims, resolved_axis, num_resolved_axis, in Mean()
198 for (int idx = 0; idx < num_resolved_axis; ++idx) { in Mean()
357 int num_resolved_axis = 0; in QuantizedMeanOrSum() local
359 &num_resolved_axis)) { in QuantizedMeanOrSum()
364 output_num_dims, resolved_axis, num_resolved_axis, in QuantizedMeanOrSum()
[all …]
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 328 int num_resolved_axis = 0; in Eval() local
330 &resolved_axis[0], &num_resolved_axis)) { in Eval()
335 num_resolved_axis, output); in Eval()
|