Searched defs:output_num_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 36 const int output_num_dims, const int* axis, in Reduce()
95 const int output_num_dims, const int* axis, in ReduceSumImpl()
130 const int* output_dims, const int output_num_dims, in ReduceGeneric()
164 const int* output_dims, const int output_num_dims, in Mean()
328 const int output_num_dims, const int* axis, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to.cc | 54 int output_num_dims = SizeOfDimension(op_context->shape, 0); in ResizeOutputTensor() local
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 486 int output_num_dims, int* forwarded_input, TF_Status* status) { in TF_ForwardInputOrAllocateOutput()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1199 const int* output_dims, const int output_num_dims, in MeanGeneral()
|