Searched refs:type_param (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/ |
| D | lsh_projection_weights_as_inputs.mod.py | 26 type_param = Input("type_param", "TENSOR_INT32", "{1}") variable
28 model = model.Operation("LSH_PROJECTION", hhash, lookup, weight, type_param).To(output)
33 type_param: [2], # DENSE
|
| D | lsh_projection_2.mod.py | 27 type_param = Int32Scalar("type_param", 1) # SPARSE variable
30 type_param).To(output)
|
| D | lsh_projection.mod.py | 27 type_param = Int32Scalar("type_param", 2) # DENSE variable
30 type_param).To(output)
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | lsh_projection_weights_as_inputs.model.cpp | 12 auto type_param = model->addOperand(&type3); in CreateModel() local
15 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output}); in CreateModel()
18 {hash, lookup, weight, type_param}, in CreateModel()
|
| D | lsh_projection.model.cpp | 12 auto type_param = model->addOperand(&type3); in CreateModel() local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1); in CreateModel()
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output}); in CreateModel()
|
| D | lsh_projection_2.model.cpp | 12 auto type_param = model->addOperand(&type3); in CreateModel() local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1); in CreateModel()
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output}); in CreateModel()
|