Searched refs:ResolveAxis (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 58 inline bool ResolveAxis(const int num_dims, const int* axis, in ResolveAxis() function
148 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
185 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in Mean()
358 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 288 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis() function
307 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMeanReferenceOps()
402 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMean()
421 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMean()
450 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMean()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2020-009.md | 7 To mimic Python's indexing with negative values, TFLite uses `ResolveAxis` to
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 95 inline bool ResolveAxis(const int num_dims, const int* axis, const int num_axis, in ResolveAxis() function
329 if (!ResolveAxis(input->dims->size, axis->data.i32, num_axis, in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reduce.cc | 127 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis() function
152 ResolveAxis(tflite::micro::GetTensorData<int>(axis), num_axis, &op_params); in EvalMean()
|