Searched defs:key_tensor (Results 1 – 7 of 7) sorted by relevance
42 const TfLiteTensor* key_tensor; in PrepareHashtableImport() local64 const TfLiteTensor* key_tensor; in EvalHashtableImport() local
47 const TfLiteTensor* key_tensor; in PrepareHashtableFind() local68 const TfLiteTensor* key_tensor; in EvalHashtableFind() local
722 TfLiteTensor key_tensor = CreateTensor<KeyType>(key_type, keys); in InitHashtableResource() local752 auto key_tensor = interpreter_->tensor(keys_); in CreateHashtableResource() local
526 const Tensor* key_tensor; in Compute() local578 const Tensor* key_tensor; in Compute() local627 const Tensor* key_tensor; in Compute() local
248 key_tensor, argument
92 Tensor& key_tensor = out_tensors->back(); in GetNextInternal() local
488 Tensor key_tensor = ctx->input(ctx->num_inputs() - 1); in Compute() local