Searched refs:out_indices (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | topk_v2.mod.py | 30 out_indices=Output("out_indices", "TENSOR_INT32", "{2, 2}"),
38 out_indices=Output("out_indices", "TENSOR_INT32", "{2, 2}"),
46 out_indices=Output("out_indices", "TENSOR_INT32", "{2, 2}"),
54 out_indices=Output("out_indices", "TENSOR_INT32", "{2}"),
62 out_indices=Output("out_indices", "TENSOR_INT32", "{2, 2}"),
70 out_indices=Output("out_indices", "TENSOR_INT32", "{2, 2}"),
76 test_case.out_values, test_case.out_indices)
80 test_case.out_indices: test_case.out_indices_data
|
| /frameworks/base/libs/androidfw/include/androidfw/ |
| D | AttributeResolution.h | 50 size_t attrs_length, uint32_t* out_values, uint32_t* out_indices);
56 uint32_t* out_values, uint32_t* out_indices);
61 size_t attrs_length, uint32_t* out_values, uint32_t* out_indices);
|
| /frameworks/base/libs/androidfw/ |
| D | AttributeResolution.cpp | 66 size_t attrs_length, uint32_t* out_values, uint32_t* out_indices) { in ResolveAttrs() argument
190 if (out_indices != nullptr && in ResolveAttrs()
193 out_indices[indices_idx] = ii; in ResolveAttrs()
199 if (out_indices != nullptr) { in ResolveAttrs()
200 out_indices[0] = indices_idx; in ResolveAttrs()
207 uint32_t* out_values, uint32_t* out_indices) { in ApplyStyle() argument
397 out_indices[indices_idx] = ii; in ApplyStyle()
403 out_indices[0] = indices_idx; in ApplyStyle()
407 size_t attrs_length, uint32_t* out_values, uint32_t* out_indices) { in RetrieveAttributes() argument
467 if (out_indices != nullptr && in RetrieveAttributes()
[all …]
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | topk_v2.model.cpp | 11 auto out_indices = model->addOperand(&type2); in CreateModel() local
15 model->addOperation(ANEURALNETWORKS_TOPK_V2, {input, k}, {out_values, out_indices}); in CreateModel()
19 {out_values, out_indices}); in CreateModel()
36 auto out_indices = model->addOperand(&type2); in CreateModel_relaxed() local
40 model->addOperation(ANEURALNETWORKS_TOPK_V2, {input, k}, {out_values, out_indices}); in CreateModel_relaxed()
44 {out_values, out_indices}); in CreateModel_relaxed()
63 auto out_indices = model->addOperand(&type2); in CreateModel_float16() local
67 model->addOperation(ANEURALNETWORKS_TOPK_V2, {input, k}, {out_values, out_indices}); in CreateModel_float16()
71 {out_values, out_indices}); in CreateModel_float16()
89 auto out_indices = model->addOperand(&type13); in CreateModel_dynamic_output_shape() local
[all …]
|
| /frameworks/base/core/jni/ |
| D | android_util_AssetManager.cpp | 1232 uint32_t* out_indices = reinterpret_cast<uint32_t*>(out_indices_ptr); in NativeApplyStyle() local
1242 out_values, out_indices); in NativeApplyStyle()
1283 jint* out_indices = nullptr; in NativeResolveAttrs() local
1287 out_indices = in NativeResolveAttrs()
1289 if (out_indices == nullptr) { in NativeResolveAttrs()
1308 attrs_len, reinterpret_cast<uint32_t*>(out_values), reinterpret_cast<uint32_t*>(out_indices)); in NativeResolveAttrs()
1309 if (out_indices != nullptr) { in NativeResolveAttrs()
1310 env->ReleasePrimitiveArrayCritical(out_java_indices, out_indices, 0); in NativeResolveAttrs()
1343 jint* out_indices = nullptr; in NativeRetrieveAttributes() local
1347 out_indices = in NativeRetrieveAttributes()
[all …]
|