Lines Matching refs:operands
34 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) { in LSHProjection() argument
35 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
36 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
37 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
40 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
42 output_ = GetOutput(operation, operands, kOutputTensor); in LSHProjection()
45 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare() argument
50 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
55 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
60 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
63 const auto& typeOperand = operands[operation.inputs[kTypeParam]]; in Prepare()
69 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare()
73 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
74 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()