Searched refs:input_num_dims (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | squeeze.cc | 44 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 D | mean.cc | 72 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 D | reference_ops.h | 2798 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