Searched defs:output_num_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 57 const int output_num_dims, const int* axis, in Reduce()
82 const int output_num_dims, const int* axis, in Reduce()
146 const int output_num_dims, const int* axis, in ReduceSumImpl()
182 const int* output_dims, const int output_num_dims, in ReduceGeneric()
219 const int* output_dims, const int output_num_dims, in Mean()
383 const int output_num_dims, const int* axis, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 406 const int* output_dims, const int output_num_dims, in QuantizedMeanOrSum()
497 int output_num_dims, T* output_data, in ReduceDispatcher()
673 const int* output_dims, const int output_num_dims, in MeanGeneral()
731 const int* output_dims, const int output_num_dims, in Mean()
781 const int* output_dims, const int output_num_dims, in ReduceGeneric()
|
| /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 | 709 const int64_t* output_dims, int output_num_dims, int* forwarded_input, in TF_ForwardInputOrAllocateOutput()
|