Searched refs:keep_dims (Results 1 – 6 of 6) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | reduce.cc | 163 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
176 params->keep_dims, temp_index, resolved_axis, in EvalMean()
182 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
200 params->keep_dims, temp_index, resolved_axis, temp_buffer)); in EvalMean()
212 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
218 if (params->keep_dims && special_case_4d_axes_1_and_2) { in EvalMean()
236 num_axis, params->keep_dims, temp_index, in EvalMean()
249 num_axis, params->keep_dims, temp_index, resolved_axis, in EvalMean()
285 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
302 params->keep_dims, temp_buffer, resolved_axis, in EvalMax()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
| A D | reduce.h | 132 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric() argument
165 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean() argument
329 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum() argument
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/ |
| A D | builtin_op_data.h | 376 bool keep_dims; member
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/ |
| A D | flatbuffer_conversions.cc | 1651 params->keep_dims = schema_params->keep_dims(); in ParseReducer()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/ |
| A D | schema_generated.h | 6682 bool keep_dims;
6684 : keep_dims(false) {
6693 bool keep_dims() const {
6709 void add_keep_dims(bool keep_dims) {
6710 fbb_.AddElement<uint8_t>(ReducerOptions::VT_KEEP_DIMS, static_cast<uint8_t>(keep_dims), 0);
6726 bool keep_dims = false) {
6728 builder_.add_keep_dims(keep_dims);
13247 { auto _e = keep_dims(); _o->keep_dims = _e; }
13258 auto _keep_dims = _o->keep_dims;
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
| A D | optimized_ops.h | 1199 bool keep_dims, int* temp_index, int* resolved_axis, in MeanGeneral() argument
1203 num_axis_dimensions, keep_dims, temp_index, in MeanGeneral()
1211 const int* axis, const int num_axis_dimensions, bool keep_dims,
1232 num_axis_dimensions, keep_dims, temp_index,
|
Completed in 51 milliseconds