Lines Matching refs:context

47 bool execute(IOperationExecutionContext* context, float func(float)) {  in execute()  argument
48 switch (context->getInputType(kInputTensor)) { in execute()
50 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
51 context->getInputShape(kInputTensor), in execute()
52 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
54 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
55 context->getInputShape(kInputTensor), in execute()
56 context->getOutputBuffer<float>(kOutputTensor)); in execute()
64 bool executeAbs(IOperationExecutionContext* context) { in executeAbs() argument
65 switch (context->getInputType(kInputTensor)) { in executeAbs()
68 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
69 context->getInputShape(kInputTensor), in executeAbs()
70 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
72 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
73 context->getInputShape(kInputTensor), in executeAbs()
74 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
77 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
78 context->getInputShape(kInputTensor), in executeAbs()
79 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
85 Result<Version> validate(const IOperationValidationContext* context) { in validate() argument
86 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
87 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
88 OperandType inputType = context->getInputType(kInputTensor); in validate()
92 NN_RET_CHECK(validateInputTypes(context, {inputType})); in validate()
93 NN_RET_CHECK(validateOutputTypes(context, {inputType})); in validate()
97 Result<Version> validateAbs(const IOperationValidationContext* context) { in validateAbs() argument
98 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateAbs()
99 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateAbs()
100 OperandType inputType = context->getInputType(kInputTensor); in validateAbs()
104 NN_RET_CHECK(validateInputTypes(context, {inputType})); in validateAbs()
105 NN_RET_CHECK(validateOutputTypes(context, {inputType})); in validateAbs()
109 Result<Version> validateFloor(const IOperationValidationContext* context) { in validateFloor() argument
110 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateFloor()
111 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateFloor()
113 OperandType inputType = context->getInputType(kInputTensor); in validateFloor()
117 NN_RET_CHECK(validateInputTypes(context, {inputType})); in validateFloor()
118 NN_RET_CHECK(validateOutputTypes(context, {inputType})); in validateFloor()
120 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
128 bool prepare(IOperationExecutionContext* context) { in prepare() argument
129 Shape input = context->getInputShape(kInputTensor); in prepare()
130 Shape output = context->getOutputShape(kOutputTensor); in prepare()
132 return context->setOutputShape(kOutputTensor, output); in prepare()
135 bool prepareFloor(IOperationExecutionContext* context) { in prepareFloor() argument
136 Shape input = context->getInputShape(kInputTensor); in prepareFloor()
137 Shape output = context->getOutputShape(kOutputTensor); in prepareFloor()
140 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
143 bool executeExp(IOperationExecutionContext* context) { in executeExp() argument
144 return execute(context, std::exp); in executeExp()
147 bool executeFloor(IOperationExecutionContext* context) { in executeFloor() argument
148 return execute(context, std::floor); in executeFloor()
151 bool executeLog(IOperationExecutionContext* context) { in executeLog() argument
152 return execute(context, std::log); in executeLog()
155 bool executeRsqrt(IOperationExecutionContext* context) { in executeRsqrt() argument
156 return execute(context, [](float x) { return 1.f / std::sqrt(x); }); in executeRsqrt()
159 bool executeSin(IOperationExecutionContext* context) { in executeSin() argument
160 return execute(context, std::sin); in executeSin()
163 bool executeSqrt(IOperationExecutionContext* context) { in executeSqrt() argument
164 return execute(context, std::sqrt); in executeSqrt()