| /external/tensorflow/tensorflow/core/kernels/ |
| lookup_table_op.cc | 119 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
248 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
488 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
576 key_dtype(), TensorShape({num_buckets_, key_size}), &key_buckets_,
662 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
698 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
775 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
801 #define REGISTER_KERNEL(key_dtype, value_dtype) \
805 .TypeConstraint<key_dtype>("key_dtype") \
[all...] |
| lookup_util.cc | 76 DataType key_dtype, int64 key_index, DataType value_dtype,
84 key_ = Tensor(key_dtype, TensorShape({}));
314 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype,
316 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) {
318 "Conflicting key/value dtypes ", key_dtype, "->", value_dtype, " with ",
319 table.key_dtype(), "-", table.value_dtype(), " for table ", table_name);
329 if (key_index == kLineNumber && table->key_dtype() != DT_INT64) {
332 table->key_dtype());
334 const DataType& key_dtype = table->key_dtype() local
[all...] |
| lookup_util.h | 39 // Verify that the given key_dtype and value_dtype matches the corresponding
41 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype,
|
| lookup_table_op.h | 36 // key and value are of the templated type "key_dtype" and "value_dtype"
38 template <class Container, class key_dtype, class value_dtype>
82 *table, DataTypeToEnum<key_dtype>::v(),
178 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
|
| lookup_table_init_op.cc | 55 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
|
| /external/tensorflow/tensorflow/python/ops/ |
| lookup_ops.py | 78 def _check_table_dtypes(table, key_dtype, value_dtype):
79 """Check that the given key_dtype and value_dtype matches the table dtypes.
83 key_dtype: The key data type to check.
87 TypeError: when 'key_dtype' or 'value_dtype' doesn't match the table data
90 if key_dtype.base_dtype != table.key_dtype:
92 (table.key_dtype, key_dtype))
101 def __init__(self, key_dtype, value_dtype, name):
105 key_dtype: The table key type
114 def key_dtype(self): member in class:LookupInterface
302 def key_dtype(self): member in class:TableInitializerBase
[all...] |
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| api_def_MutableHashTableOfTensors.pbtxt | 25 name: "key_dtype"
|
| api_def_MutableHashTableOfTensorsV2.pbtxt | 27 name: "key_dtype"
|
| api_def_HashTable.pbtxt | 32 name: "key_dtype"
|
| api_def_HashTableV2.pbtxt | 34 name: "key_dtype"
|
| api_def_MutableHashTable.pbtxt | 32 name: "key_dtype"
|
| api_def_MutableHashTableV2.pbtxt | 34 name: "key_dtype"
|
| api_def_MutableDenseHashTable.pbtxt | 32 name: "key_dtype"
|
| api_def_MutableDenseHashTableV2.pbtxt | 34 name: "key_dtype"
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| lookup_ops.py | 56 key_dtype=dtypes.string, name=name)
298 table = tf.contrib.lookup.MutableHashTable(key_dtype=tf.string,
308 key_dtype,
316 Creates a table, the type of its keys and values are specified by key_dtype
320 key_dtype: the type of the key tensors.
349 key_dtype=key_dtype,
356 key_dtype=key_dtype,
361 super(MutableHashTable, self).__init__(key_dtype, value_dtype
[all...] |
| lookup_ops_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| sharded_mutable_dense_hashtable.py | 47 key_dtype,
54 super(ShardedMutableDenseHashTable, self).__init__(key_dtype,
60 key_dtype=key_dtype,
|
| /external/tensorflow/tensorflow/core/framework/ |
| lookup_interface.h | 84 virtual DataType key_dtype() const = 0;
98 // - DataType of the tensor keys equals to the table key_dtype
112 // - DataType of the tensor keys equals to the table key_dtype
131 // key_dtype and value_dtype.
|
| lookup_interface.cc | 35 if (keys.dtype() != key_dtype()) {
36 return errors::InvalidArgument("Key must be type ", key_dtype(),
|
| /external/tensorflow/tensorflow/core/ops/ |
| lookup_ops.cc | 222 .Attr("key_dtype: type")
232 .Attr("key_dtype: type")
242 .Attr("key_dtype: type")
252 .Attr("key_dtype: type")
262 .Attr("key_dtype: type")
273 .Attr("key_dtype: type")
280 .Input("empty_key: key_dtype")
285 .Attr("key_dtype: type")
294 .Input("empty_key: key_dtype")
299 .Attr("key_dtype: type"
[all...] |
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine_test.cc | 389 const DataType key_dtype = DT_INT64; local
395 SetNodeAttr("key_dtype", key_dtype, hashtable_node);
402 SetNodeAttr("dtype", key_dtype, keys_node);
403 Tensor keys(key_dtype, TensorShape{2});
421 SetNodeAttr("Tkey", key_dtype, init_table_node);
432 SetNodeAttr("dtype", key_dtype, query_node);
433 Tensor query(key_dtype, TensorShape({}));
450 SetNodeAttr("Tin", key_dtype, lookup_node);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| lookup_ops_test.py | 351 key_dtype=dtypes.int32)
366 key_dtype=dtypes.int64)
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| saver_test_utils.py | 39 key_dtype=dtypes.string, value_dtype=dtypes.float32, name=name)
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| sparsify_gather.cc | 357 DataType key_dtype = DT_INT64;
362 SetNodeAttr("dtype", key_dtype, &indices_node);
373 SetNodeAttr("key_dtype", key_dtype, &hashtable_node);
381 SetNodeAttr("Tkey", key_dtype, &init_table_node);
389 SetNodeAttr("Tin", key_dtype, &lookup_node);
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| math_utils.py | 510 def __init__(self, key_dtype, value_dtype, default_value, *args, **kwargs):
513 key_dtype=key_dtype,
547 self, key_dtype, default_values, empty_key, name, checkpoint=True):
552 key_dtype=key_dtype,
580 def check_table_dtypes(self, key_dtype, value_dtype):
[all...] |
