Searched refs:input_num_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 35 const int* output_dims, const int input_num_dims, in Reduce() argument
41 for (int idx = 0; idx < input_num_dims; ++idx) { in Reduce()
47 ReducedOutputOffset(input_num_dims, input_dims, input_iter, 0, nullptr); in Reduce()
48 size_t output_offset = ReducedOutputOffset(input_num_dims, input_dims, in Reduce()
52 } while (NextIndex(input_num_dims, input_dims, input_iter)); in Reduce()
94 const int* output_dims, const int input_num_dims, in ReduceSumImpl() argument
102 return Reduce<In, Out>(input_data, input_dims, output_dims, input_num_dims, in ReduceSumImpl()
129 const int input_num_dims, T* output_data, in ReduceGeneric() argument
136 for (int i = 0; i < input_num_dims; ++i) { in ReduceGeneric()
148 if (!ResolveAxis(input_num_dims, axis, num_axis_dimensions, resolved_axis, in ReduceGeneric()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 43 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()
|
| D | broadcast_to.cc | 53 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
55 TF_LITE_ENSURE_MSG(context, input_num_dims <= output_num_dims, in ResizeOutputTensor()
69 int extending_dims = output_num_dims - input_num_dims; in ResizeOutputTensor()
70 for (int idx = 0; idx < input_num_dims; ++idx) { in ResizeOutputTensor()
|
| D | reduce.cc | 101 int input_num_dims = NumDimensions(op_context->input); in ResizeOutputTensor() local
102 if (input_num_dims == 0) { in ResizeOutputTensor()
108 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_num_dims); in ResizeOutputTensor()
109 for (int idx = 0; idx < input_num_dims; ++idx) { in ResizeOutputTensor()
112 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
130 current += input_num_dims; in ResizeOutputTensor()
132 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims); in ResizeOutputTensor()
136 previous += input_num_dims; in ResizeOutputTensor()
146 TfLiteIntArrayCreate(input_num_dims - num_reduce_axis); in ResizeOutputTensor()
148 for (int idx = 0; idx < input_num_dims; ++idx) { in ResizeOutputTensor()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 551 const int input_num_dims, float* output_data,
563 const int input_num_dims, int8_t* output_data,
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1741 const int input_num_dims = input_dims.size(); in ProcessSqueezeOperator() local
1743 squeeze_dims.push_back(i < 0 ? i + input_num_dims : i); in ProcessSqueezeOperator()
1745 for (int i = 0; i < input_num_dims; ++i) { in ProcessSqueezeOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1198 const int input_num_dims, T* output_data, in MeanGeneral() argument
1203 return reference_ops::Mean(input_data, input_dims, input_num_dims, in MeanGeneral()
1211 const float* input_data, const int* input_dims, const int input_num_dims,
1216 if (num_axis_dimensions == 1 && axis[0] == (input_num_dims - 1)) {
1219 for (int i = 0; i < input_num_dims - 1; ++i) {
1232 return reference_ops::Mean(input_data, input_dims, input_num_dims,
|