Searched refs:kKeyTensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_import.cc | 28 constexpr int kKeyTensor = 1; variable
44 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableImport()
66 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableImport()
|
| H A D | hashtable_find.cc | 28 constexpr int kKeyTensor = 1; variable
49 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableFind()
70 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableFind()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | HashtableLookup.h | 37 static constexpr int kKeyTensor = 1; variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HashtableLookup.cpp | 38 key_ = GetInput(operation, operands, kKeyTensor); in HashtableLookup()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 959 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()
|