Searched refs:input_num_dims (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dsqueeze.cc44 int input_num_dims = NumDimensions(op_context.input); local
50 TF_LITE_ENSURE(context, input_num_dims <= 8);
54 for (int idx = 0; idx < input_num_dims; ++idx) {
62 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims
64 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims &&
72 TfLiteIntArrayCreate(input_num_dims - num_squeezed_dims);
73 for (int in_idx = 0, out_idx = 0; in_idx < input_num_dims; ++in_idx) {
H A Dmean.cc72 int input_num_dims = NumDimensions(op_context->input); local
75 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims);
76 for (int idx = 0; idx < input_num_dims; ++idx) {
79 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) {
97 current += input_num_dims;
99 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims);
103 previous += input_num_dims;
113 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis);
115 for (int idx = 0; idx < input_num_dims; ++idx) {
118 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Dreference_ops.h2798 inline void Mean(T* input_data, const int* input_dims, const int input_num_dims,
2812 for (int idx = 0; idx < input_num_dims; ++idx) {
2819 TFLITE_DCHECK(current < input_num_dims && current + input_num_dims >= 0);
2821 current += input_num_dims;
2836 has_next = NextIndex(input_num_dims, input_dims, temp_index)) {
2838 ReducedOutputOffset(input_num_dims, input_dims, temp_index, 0, nullptr);
2840 ReducedOutputOffset(input_num_dims, input_dims, temp_index,

Completed in 182 milliseconds