| /external/tensorflow/tensorflow/lite/core/api/ |
| flatbuffer_conversions.cc | 39 int num_dimensions = flat_vector->Length(); local
40 if (num_dimensions > max_size_of_buffer / sizeof(int)) {
46 for (int i = 0; i < num_dimensions; ++i) {
463 params->num_dimensions = new_shape->Length();
|
| /art/runtime/mirror/ |
| array.cc | 86 int num_dimensions = dimensions->GetLength(); local
87 DCHECK_GT(num_dimensions, 0);
88 DCHECK_LE(num_dimensions, 255);
90 for (int i = 0; i < num_dimensions; i++) {
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| gather.cc | 80 const int num_dimensions = local
82 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions);
|
| reshape.cc | 80 int num_dimensions = params->num_dimensions; local
81 if (num_dimensions == 1 && params->shape[0] == 0) {
85 num_dimensions = 0;
87 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions);
88 for (int i = 0; i < num_dimensions; ++i) {
|
| tile.cc | 36 int num_dimensions) {
39 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions);
40 for (int i = 0; i < num_dimensions; ++i) {
51 const int num_dimensions = NumDimensions(input); local
53 TF_LITE_ENSURE_EQ(context, num_dimensions, num_multipliers);
59 num_dimensions));
64 num_dimensions));
|
| topk_v2.cc | 41 const int num_dimensions = NumDimensions(input); local
46 TF_LITE_ENSURE_MSG(context, k <= input->dims->data[num_dimensions - 1],
49 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions);
50 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions);
51 for (int i = 0; i < num_dimensions - 1; ++i) {
55 output_indexes_shape->data[num_dimensions - 1] = k;
56 output_values_shape->data[num_dimensions - 1] = k;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_conv_runner.cc | 139 const int num_dimensions = window.dimensions_size(); local
140 CHECK_LE(num_dimensions, 3);
141 CHECK_GE(num_dimensions, 1);
146 const int effective_num_dimensions = std::max(2, num_dimensions);
156 CHECK_EQ(num_dimensions, dnums.input_spatial_dimensions_size());
157 CHECK_EQ(num_dimensions, dnums.kernel_spatial_dimensions_size());
158 CHECK_EQ(num_dimensions, dnums.output_spatial_dimensions_size());
186 for (int dim = 0; dim < num_dimensions; ++dim) {
199 for (int dim = 0; dim < num_dimensions; ++dim) {
208 for (int dim = 0; dim < num_dimensions; ++dim)
[all...] |
| /external/tensorflow/tensorflow/lite/c/ |
| builtin_op_data.h | 251 int num_dimensions; member in struct:__anon45870
|
