Searched defs:key_tensor (Results 1 – 7 of 7) sorted by relevance
44 const TfLiteTensor* key_tensor; in PrepareHashtableImport() local66 const TfLiteTensor* key_tensor; in EvalHashtableImport() local
50 const TfLiteTensor* key_tensor; in PrepareHashtableFind() local71 const TfLiteTensor* key_tensor; in EvalHashtableFind() local
737 TfLiteTensor key_tensor = CreateTensor<KeyType>(key_type, keys); in InitHashtableResource() local768 auto key_tensor = interpreter_->tensor(keys_); in CreateHashtableResource() local
523 const Tensor* key_tensor; in Compute() local573 const Tensor* key_tensor; in Compute() local622 const Tensor* key_tensor; in Compute() local
252 key_tensor, argument
92 Tensor& key_tensor = out_tensors->back(); in GetNextInternal() local
460 Tensor key_tensor = ctx->input(ctx->num_inputs() - 1); in Compute() local