Searched refs:key_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_import.cc | 44 const TfLiteTensor* key_tensor; in PrepareHashtableImport() local
46 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableImport()
50 TF_LITE_ENSURE(context, (key_tensor->type == kTfLiteInt64 && in PrepareHashtableImport()
52 (key_tensor->type == kTfLiteString && in PrepareHashtableImport()
56 TF_LITE_ENSURE(context, HaveSameShapes(key_tensor, value_tensor)); in PrepareHashtableImport()
66 const TfLiteTensor* key_tensor; in EvalHashtableImport() local
68 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableImport()
78 lookup->CheckKeyAndValueTypes(context, key_tensor, value_tensor)); in EvalHashtableImport()
81 auto result = lookup->Import(context, key_tensor, value_tensor); in EvalHashtableImport()
|
| D | hashtable_find.cc | 50 const TfLiteTensor* key_tensor; in PrepareHashtableFind() local
52 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableFind()
57 TF_LITE_ENSURE(context, (key_tensor->type == kTfLiteInt64 && in PrepareHashtableFind()
59 (key_tensor->type == kTfLiteString && in PrepareHashtableFind()
62 TfLiteIntArrayCopy(key_tensor->dims)); in PrepareHashtableFind()
71 const TfLiteTensor* key_tensor; in EvalHashtableFind() local
73 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableFind()
85 lookup->CheckKeyAndValueTypes(context, key_tensor, output_tensor)); in EvalHashtableFind()
87 lookup->Lookup(context, key_tensor, output_tensor, default_value_tensor); in EvalHashtableFind()
|
| D | hashtable_ops_test.cc | 737 TfLiteTensor key_tensor = CreateTensor<KeyType>(key_type, keys); in InitHashtableResource() local
739 lookup->Import(&context, &key_tensor, &value_tensor); in InitHashtableResource()
740 TfLiteTensorFree(&key_tensor); in InitHashtableResource()
768 auto key_tensor = interpreter_->tensor(keys_); in CreateHashtableResource() local
773 &resources, resource_id, key_tensor->type, value_tensor->type); in CreateHashtableResource()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | map_stage_op.cc | 523 const Tensor* key_tensor; in Compute() local
527 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
532 Tensor key(*key_tensor); in Compute()
573 const Tensor* key_tensor; in Compute() local
576 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
578 OP_REQUIRES_OK(ctx, map->pop(key_tensor, indices_tensor, &tuple)); in Compute()
622 const Tensor* key_tensor; in Compute() local
625 OP_REQUIRES_OK(ctx, ctx->input("key", &key_tensor)); in Compute()
627 OP_REQUIRES_OK(ctx, map->get(key_tensor, indices_tensor, &tuple)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | lmdb_dataset_op.cc | 92 Tensor& key_tensor = out_tensors->back(); in GetNextInternal() local
93 key_tensor.scalar<tstring>()() = string( in GetNextInternal()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 224 key_tensor = keys
227 key_tensor = keys.values
235 (self.resource_handle, key_tensor, self._default_value)):
237 key_tensor,
240 values.set_shape(key_tensor.get_shape())
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 460 Tensor key_tensor = ctx->input(ctx->num_inputs() - 1); in Compute() local
461 const XlaExecutableClosureStore::KeyT& key = key_tensor.flat<tstring>()(0); in Compute()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 252 key_tensor, argument
|