Searched defs:input_num_dims (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reduce.h | 56 const int* output_dims, const int input_num_dims, in Reduce()
81 const int* output_dims, const int input_num_dims, in Reduce()
145 const int* output_dims, const int input_num_dims, in ReduceSumImpl()
181 const int input_num_dims, T* output_data, in ReduceGeneric()
218 const int input_num_dims, T* output_data, in Mean()
380 const int input_num_dims, T* output_data, in QuantizedMeanOrSum()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce.h | 313 const int input_num_dims, T* output_data) { in ReduceIsCopy()
388 const int input_num_dims, const int* axis, in Reduce()
404 const int input_num_dims, T* output_data, in QuantizedMeanOrSum()
496 const int input_num_dims, const int* output_dims, in ReduceDispatcher()
672 const int input_num_dims, T* output_data, in MeanGeneral()
730 const int input_num_dims, T* output_data, in Mean()
780 const int input_num_dims, T* output_data, in ReduceGeneric()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | squeeze.cc | 43 int input_num_dims = NumDimensions(op_context.input); in Prepare() local
|
| H A D | broadcast_to.cc | 53 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
|
| H A D | reduce.cc | 104 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
653 const int input_num_dims, T* output_data, T init_value, in ReduceAllDims()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 1741 const int input_num_dims = input_dims.size(); in ProcessSqueezeOperator() local
|