Searched refs:LSHProjection (Results 1 – 5 of 5) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSHProjection.cpp | 33 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) { in LSHProjection() function in android::nn::LSHProjection
44 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
160 bool LSHProjection::Eval() { in Eval()
179 template bool LSHProjection::Eval<float>();
180 template bool LSHProjection::Eval<_Float16>();
|
| D | LSHProjectionTest.cpp | 107 execution.setInput(LSHProjection::k##X##Tensor, X##_.data(), sizeof(T) * X##_.size()), \ in Invoke()
115 ASSERT_EQ(execution.setOutput(LSHProjection::k##X##Tensor, X##_.data(), \ in Invoke()
123 ASSERT_EQ(execution.setInput(LSHProjection::kTypeParam, &type_, sizeof(type_)), in Invoke()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSHProjection.h | 38 class LSHProjection {
40 LSHProjection(const Operation& operation, RunTimeOperandInfo* operands);
|
| /packages/modules/NeuralNetworks/common/ |
| D | Android.bp | 156 "cpu_operations/LSHProjection.cpp",
343 "cpu_operations/LSHProjection.cpp",
|
| D | CpuExecutor.cpp | 974 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
976 if (!LSHProjection::Prepare(operation, operands, &outputShape) || in executeOperation()
981 LSHProjection lsh(operation, operands); in executeOperation()
982 const RunTimeOperandInfo& hash = operands[ins[LSHProjection::kHashTensor]]; in executeOperation()
|