Home
last modified time | relevance | path

Searched refs:key_dtype (Results 1 – 25 of 87) sorted by relevance

1234

/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py89 def _check_table_dtypes(table, key_dtype, value_dtype): argument
101 if key_dtype.base_dtype != table.key_dtype:
103 (table.key_dtype, key_dtype))
112 def __init__(self, key_dtype, value_dtype): argument
119 self._key_dtype = dtypes.as_dtype(key_dtype)
127 def key_dtype(self): member in LookupInterface
171 super(InitializableLookupTableBase, self).__init__(initializer.key_dtype,
320 key_dtype=self._initializer.key_dtype,
402 def __init__(self, key_dtype, value_dtype): argument
409 self._key_dtype = dtypes.as_dtype(key_dtype)
[all …]
Dmap_ops.py50 def tensor_map_stack_keys(input_handle, key_dtype): argument
51 return gen_map_ops.tensor_map_stack_keys(input_handle, key_dtype)
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable.cc40 TfLiteType key_dtype; member
51 TfLiteType key_dtype, value_dtype; in InitHashtable() local
53 &key_dtype, nullptr); in InitHashtable()
59 option->key_dtype = key_dtype; in InitHashtable()
78 TF_LITE_ENSURE(context, (params->key_dtype == kTfLiteInt64 && in PrepareHashtable()
80 (params->key_dtype == kTfLiteString && in PrepareHashtable()
121 &resources, resource_id, params->key_dtype, params->value_dtype); in EvalHashtable()
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc79 DataType key_dtype, int64 key_index, DataType value_dtype, in Init() argument
84 key_ = Tensor(key_dtype, TensorShape({})); in Init()
343 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes() argument
345 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) { in CheckTableDataTypes()
347 "Conflicting key/value dtypes ", DataTypeString(key_dtype), "->", in CheckTableDataTypes()
349 DataTypeString(table.key_dtype()), "-", in CheckTableDataTypes()
360 if (key_index == kLineNumber && table->key_dtype() != DT_INT64) { in InitializeTableFromTextFile()
363 DataTypeString(table->key_dtype())); in InitializeTableFromTextFile()
365 const DataType& key_dtype = table->key_dtype(); in InitializeTableFromTextFile() local
367 if (key_index == kWholeLine && !DataTypeIsInteger(key_dtype) && in InitializeTableFromTextFile()
[all …]
Dlookup_table_op.cc139 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
288 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
574 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableDenseHashTable
728 key_dtype(), TensorShape({num_buckets_, key_size}), &key_buckets_, in AllocateBuckets()
834 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute()
868 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute()
903 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype()}; in Compute()
974 DataTypeVector expected_inputs = {expected_input_0_, table->key_dtype(), in Compute()
1000 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1004 .TypeConstraint<key_dtype>("key_dtype") \
[all …]
/external/tensorflow/tensorflow/core/framework/
Dlookup_interface.cc35 if (keys.dtype() != key_dtype()) { in CheckKeyAndValueTypes()
36 return errors::InvalidArgument("Key must be type ", key_dtype(), in CheckKeyAndValueTypes()
75 if (keys.dtype() != key_dtype()) { in CheckKeyTensorForRemove()
76 return errors::InvalidArgument("Key must be type ", key_dtype(), in CheckKeyTensorForRemove()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DMutableDenseHashTableV2.pbtxt5 type_attr: "key_dtype"
9 type_attr: "key_dtype"
37 name: "key_dtype"
DMutableDenseHashTable.pbtxt5 type_attr: "key_dtype"
34 name: "key_dtype"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DMutableDenseHashTableV2.pbtxt5 type_attr: "key_dtype"
9 type_attr: "key_dtype"
37 name: "key_dtype"
DTensorMapStackKeys.pbtxt9 type_attr: "key_dtype"
12 name: "key_dtype"
DTensorMapHasKey.pbtxt9 type_attr: "key_dtype"
16 name: "key_dtype"
DTensorMapErase.pbtxt9 type_attr: "key_dtype"
16 name: "key_dtype"
DTensorMapLookup.pbtxt9 type_attr: "key_dtype"
16 name: "key_dtype"
DTensorMapInsert.pbtxt9 type_attr: "key_dtype"
20 name: "key_dtype"
DMutableDenseHashTable.pbtxt5 type_attr: "key_dtype"
34 name: "key_dtype"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dinit_text_file_to_import_invalid.mlir7 …%0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf.string, shared_name = "hash_…
19 …ensor<!tf.string> %0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf.string, sh…
33 …%0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf.string, shared_name = "hash_…
Dinit_text_file_to_import.mlir7 …%0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf.string, shared_name = "hash_…
21 …%0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf.string, shared_name = "hash_…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
Dtensorflow.lookup.-text-file-initializer.pbtxt8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
Dtensorflow.lookup.experimental.-dense-hash-table.pbtxt10 name: "key_dtype"
27 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
Dtensorflow.lookup.-text-file-initializer.pbtxt8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc412 const DataType key_dtype = DT_INT64; in TEST_F() local
418 SetNodeAttr("key_dtype", key_dtype, hashtable_node); in TEST_F()
425 SetNodeAttr("dtype", key_dtype, keys_node); in TEST_F()
426 Tensor keys(key_dtype, TensorShape{2}); in TEST_F()
444 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
455 SetNodeAttr("dtype", key_dtype, query_node); in TEST_F()
456 Tensor query(key_dtype, TensorShape({})); in TEST_F()
473 SetNodeAttr("Tin", key_dtype, lookup_node); in TEST_F()
/external/tensorflow/tensorflow/lite/experimental/resource/
Dstatic_hashtable.cc97 TfLiteType key_dtype, in CreateHashtableResourceIfNotAvailable() argument
102 auto* hashtable = internal::CreateStaticHashtable(key_dtype, value_dtype); in CreateHashtableResourceIfNotAvailable()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dforce_shared_name_for_resource_ops.pbtxt7 key: "key_dtype"
111 key: "key_dtype"

1234