Lines Matching refs:operands
33 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) { in LSHProjection() argument
34 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
35 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
36 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
39 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
41 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
44 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare() argument
49 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
54 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
59 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
62 const auto& typeOperand = operands[operation.inputs[kTypeParam]]; in Prepare()
68 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare()
72 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
73 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()