Searched refs:type_param (Results 1 - 12 of 12) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_0/
H A Dlsh_projection.mod.py27 type_param = Int32Scalar("type_param", 2) # DENSE variable
30 type_param).To(output)
H A Dlsh_projection_2.mod.py27 type_param = Int32Scalar("type_param", 1) # SPARSE variable
30 type_param).To(output)
H A Dlsh_projection_weights_as_inputs.mod.py26 type_param = Int32Scalar("type_param", 2) # DENSE variable
28 model = model.Operation("LSH_PROJECTION", hhash, lookup, weight, type_param).To(output)
/frameworks/ml/nn/runtime/test/generated/models/
H A Dlsh_projection_weights_as_inputs.model.cpp12 auto type_param = model->addOperand(&type3); local
16 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
H A Dlsh_projection_weights_as_inputs_relaxed.model.cpp12 auto type_param = model->addOperand(&type3); local
16 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
H A Dlsh_projection.model.cpp12 auto type_param = model->addOperand(&type3); local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
H A Dlsh_projection_2.model.cpp12 auto type_param = model->addOperand(&type3); local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
H A Dlsh_projection_2_relaxed.model.cpp12 auto type_param = model->addOperand(&type3); local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
H A Dlsh_projection_relaxed.model.cpp12 auto type_param = model->addOperand(&type3); local
18 model->setOperandValue(type_param, type_param_init, sizeof(int32_t) * 1);
19 model->addOperation(ANEURALNETWORKS_LSH_PROJECTION, {hash, lookup, weight, type_param}, {output});
/frameworks/ml/nn/runtime/test/specs/V1_1/
H A Dlsh_projection_2_relaxed.mod.py27 type_param = Int32Scalar("type_param", 1) # SPARSE variable
30 type_param).To(output)
H A Dlsh_projection_relaxed.mod.py27 type_param = Int32Scalar("type_param", 2) # DENSE variable
30 type_param).To(output)
H A Dlsh_projection_weights_as_inputs_relaxed.mod.py26 type_param = Int32Scalar("type_param", 2) # DENSE variable
28 model = model.Operation("LSH_PROJECTION", hhash, lookup, weight, type_param).To(output)

Completed in 166 milliseconds