Searched defs:resolved_axis (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 83 TfLiteTensor* resolved_axis) { in ResizeTempAxis()
185 TfLiteTensor* resolved_axis; in InitializeTemporaries() local
231 TfLiteTensor* resolved_axis; in PrepareSimple() local
303 TfLiteTensor* resolved_axis, in EvalMeanReferenceOps()
376 TfLiteTensor* resolved_axis; in EvalMean() local
533 TfLiteTensor* resolved_axis; in EvalLogic() local
678 TfLiteTensor* resolved_axis; in EvalSum() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reduce.cc | 149 int resolved_axis[kMaxNumberOfReducedAxis]; in EvalMean() local
274 int* resolved_axis = static_cast<int*>( in EvalMax() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 132 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
166 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean()
330 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum()
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 326 std::vector<int> resolved_axis(num_axis); in Eval() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1201 bool keep_dims, int* temp_index, int* resolved_axis, in MeanGeneral()
|