Searched refs:output_num_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to.cc | 54 int output_num_dims = SizeOfDimension(op_context->shape, 0); in ResizeOutputTensor() local
55 TF_LITE_ENSURE_MSG(context, input_num_dims <= output_num_dims, in ResizeOutputTensor()
57 TF_LITE_ENSURE_MSG(context, output_num_dims <= kMaxDims, in ResizeOutputTensor()
69 int extending_dims = output_num_dims - input_num_dims; in ResizeOutputTensor()
78 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(output_num_dims); in ResizeOutputTensor()
81 for (int idx = 0; idx < output_num_dims; ++idx) { in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 36 const int output_num_dims, const int* axis, in Reduce() argument
95 const int output_num_dims, const int* axis, in ReduceSumImpl() argument
103 output_num_dims, axis, num_axis, input_iter, reducer, in ReduceSumImpl()
130 const int* output_dims, const int output_num_dims, in ReduceGeneric() argument
141 if (!InitTensorDataForReduce(output_dims, output_num_dims, init_value, in ReduceGeneric()
154 output_num_dims, resolved_axis, num_resolved_axis, in ReduceGeneric()
164 const int* output_dims, const int output_num_dims, in Mean() argument
170 for (int idx = 0; idx < output_num_dims; ++idx) { in Mean()
191 output_num_dims, resolved_axis, num_resolved_axis, in Mean()
328 const int output_num_dims, const int* axis, in QuantizedMeanOrSum() argument
[all …]
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.h | 352 int output_num_dims, int* forwarded_input, TF_Status* status);
|
| D | kernels.cc | 486 int output_num_dims, int* forwarded_input, TF_Status* status) { in TF_ForwardInputOrAllocateOutput() argument
495 reinterpret_cast<tensorflow::int64*>(output_dims), output_num_dims); in TF_ForwardInputOrAllocateOutput()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 552 const int* output_dims, const int output_num_dims,
564 const int* output_dims, const int output_num_dims,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1199 const int* output_dims, const int output_num_dims, in MeanGeneral() argument
1204 output_data, output_dims, output_num_dims, axis, in MeanGeneral()
1212 float* output_data, const int* output_dims, const int output_num_dims,
1233 output_data, output_dims, output_num_dims, axis,
|