Searched defs:input_num_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 54 const int* output_dims, const int input_num_dims, in Reduce()
79 const int* output_dims, const int input_num_dims, in Reduce()
143 const int* output_dims, const int input_num_dims, in ReduceSumImpl()
179 const int input_num_dims, T* output_data, in ReduceGeneric()
216 const int input_num_dims, T* output_data, in Mean()
378 const int input_num_dims, T* output_data, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 43 int input_num_dims = NumDimensions(op_context.input); in Prepare() local
|
| D | broadcast_to.cc | 53 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
|
| D | reduce.cc | 101 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1741 const int input_num_dims = input_dims.size(); in ProcessSqueezeOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1196 const int input_num_dims, T* output_data, in MeanGeneral()
|