Searched defs:resolved_axis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 83 TfLiteTensor* resolved_axis) { in ResizeTempAxis()
185 TfLiteTensor* resolved_axis; in InitializeTemporaries() local
232 TfLiteTensor* resolved_axis; in PrepareSimple() local
352 TfLiteTensor* resolved_axis, in EvalMeanReferenceOps()
458 TfLiteTensor* resolved_axis; in EvalMean() local
616 TfLiteTensor* resolved_axis; in EvalLogic() local
761 TfLiteTensor* resolved_axis; in EvalSum() local
826 TfLiteTensor* resolved_axis; in EvalQuantizedProd() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 182 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
219 int* temp_index, int* resolved_axis, U* temp_sum) { in Mean()
383 int* temp_index, int* resolved_axis, U* temp_sum, in QuantizedMeanOrSum()
477 int* resolved_axis, int32_t* temp_prod, in QuantizedReduceProd()
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 422 std::vector<int> resolved_axis(num_axis); in Eval() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1199 bool keep_dims, int* temp_index, int* resolved_axis, in MeanGeneral()
|