Home
last modified time | relevance | path

Searched refs:input_num_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsqueeze.cc43 int input_num_dims = NumDimensions(op_context.input); in Prepare() local
49 TF_LITE_ENSURE(context, input_num_dims <= 8); in Prepare()
53 for (int idx = 0; idx < input_num_dims; ++idx) { in Prepare()
61 int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] + input_num_dims in Prepare()
63 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims && in Prepare()
71 TfLiteIntArrayCreate(input_num_dims - num_squeezed_dims); in Prepare()
72 for (int in_idx = 0, out_idx = 0; in_idx < input_num_dims; ++in_idx) { in Prepare()
Dreduce.cc95 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
96 if (input_num_dims == 0) { in ResizeOutputTensor()
102 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
103 for (int idx = 0; idx < input_num_dims; ++idx) { in ResizeOutputTensor()
106 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
124 current += input_num_dims; in ResizeOutputTensor()
126 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims); in ResizeOutputTensor()
130 previous += input_num_dims; in ResizeOutputTensor()
140 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis); in ResizeOutputTensor()
142 for (int idx = 0; idx < input_num_dims; ++idx) { in ResizeOutputTensor()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h3434 const int* output_dims, const int input_num_dims, in Reduce() argument
3440 for (int idx = 0; idx < input_num_dims; ++idx) { in Reduce()
3446 ReducedOutputOffset(input_num_dims, input_dims, input_iter, 0, nullptr); in Reduce()
3447 size_t output_offset = ReducedOutputOffset(input_num_dims, input_dims, in Reduce()
3451 } while (NextIndex(input_num_dims, input_dims, input_iter)); in Reduce()
3486 const int* output_dims, const int input_num_dims, in ReduceSumImpl() argument
3494 return Reduce<In, Out>(input_data, input_dims, output_dims, input_num_dims, in ReduceSumImpl()
3521 const int input_num_dims, T* output_data, in ReduceGeneric() argument
3535 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
3540 return Reduce<T, T>(input_data, input_dims, output_dims, input_num_dims, in ReduceGeneric()
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1721 const int input_num_dims = input_dims.size(); in ProcessSqueezeOperator() local
1723 squeeze_dims.push_back(i < 0 ? i + input_num_dims : i); in ProcessSqueezeOperator()
1725 for (int i = 0; i < input_num_dims; ++i) { in ProcessSqueezeOperator()