| /external/tensorflow/tensorflow/core/framework/ |
| lookup_interface.cc | 35 if (keys.dtype() != key_dtype()) {
36 return errors::InvalidArgument("Key must be type ", key_dtype(),
75 if (keys.dtype() != key_dtype()) {
76 return errors::InvalidArgument("Key must be type ", key_dtype(),
|
| lookup_interface.h | 95 virtual DataType key_dtype() const = 0;
109 // - DataType of the tensor keys equals to the table key_dtype
124 // - DataType of the tensor keys equals to the table key_dtype
129 // - DataType of the tensor keys equals to the table key_dtype
148 // key_dtype and value_dtype.
|
| /external/tensorflow/tensorflow/python/ops/ |
| lookup_ops.py | 85 def _check_table_dtypes(table, key_dtype, value_dtype):
86 """Check that the given key_dtype and value_dtype matches the table dtypes.
90 key_dtype: The key data type to check.
94 TypeError: when 'key_dtype' or 'value_dtype' doesn't match the table data
97 if key_dtype.base_dtype != table.key_dtype:
99 (table.key_dtype, key_dtype))
108 def __init__(self, key_dtype, value_dtype):
112 key_dtype: The table key type
123 def key_dtype(self): member in class:LookupInterface
344 def key_dtype(self): member in class:TableInitializerBase
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| 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_util.cc | 76 DataType key_dtype, int64 key_index, DataType value_dtype,
81 key_ = Tensor(key_dtype, TensorShape({}));
326 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype,
328 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) {
330 "Conflicting key/value dtypes ", DataTypeString(key_dtype), "->",
332 DataTypeString(table.key_dtype()), "-",
343 if (key_index == kLineNumber && table->key_dtype() != DT_INT64) {
346 DataTypeString(table->key_dtype()));
348 const DataType& key_dtype = table->key_dtype() local
[all...] |
| lookup_table_op.cc | 128 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
266 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
552 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
706 key_dtype(), TensorShape({num_buckets_, key_size}), &key_buckets_,
794 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
830 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(),
867 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype()};
[all...] |
| 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(),
207 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/contrib/linear_optimizer/python/ops/ |
| sharded_mutable_dense_hashtable.py | 59 key_dtype,
67 self._key_dtype = key_dtype
75 key_dtype=key_dtype,
|
| sdca_ops.py | 162 key_dtype=dtypes.int64,
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| lookup_ops.py | 55 key_dtype=dtypes.string, name=name)
332 key_dtype=self._initializer.key_dtype,
|
| lookup_ops_test.py | 351 key_dtype=dtypes.int32)
365 key_dtype=dtypes.int64)
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| lookup_ops.cc | 102 DataType key_dtype; local
103 TF_RETURN_IF_ERROR(c->GetAttr(key_dtype_attr, &key_dtype));
104 if (key_shape_and_type.dtype != key_dtype) {
109 DataTypeString(key_dtype));
324 TF_RETURN_IF_ERROR(c->GetAttr("key_dtype", &key_t));
341 .Attr("key_dtype: type")
351 .Attr("key_dtype: type")
361 .Attr("key_dtype: type")
371 .Attr("key_dtype: type")
384 .Attr("key_dtype: type"
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| saver_test_utils.py | 39 key_dtype=dtypes.string, value_dtype=dtypes.float32, name=name)
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine_test.cc | 398 const DataType key_dtype = DT_INT64; local
404 SetNodeAttr("key_dtype", key_dtype, hashtable_node);
411 SetNodeAttr("dtype", key_dtype, keys_node);
412 Tensor keys(key_dtype, TensorShape{2});
430 SetNodeAttr("Tkey", key_dtype, init_table_node);
441 SetNodeAttr("dtype", key_dtype, query_node);
442 Tensor query(key_dtype, TensorShape({}));
459 SetNodeAttr("Tin", key_dtype, lookup_node);
|
| /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 | 538 def __init__(self, key_dtype, value_dtype, default_value, *args, **kwargs):
541 key_dtype=key_dtype,
575 key_dtype,
584 key_dtype=key_dtype,
613 def check_table_dtypes(self, key_dtype, value_dtype):
[all...] |
| state_management.py | 149 key_dtype=dtypes.int64,
|
| math_utils_test.py | 224 key_dtype=dtypes.int64,
|
| /external/tensorflow/tensorflow/contrib/text/python/ops/ |
| skip_gram_ops.py | 363 key_dtype=vocab_token_dtype,
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| load_v1_in_v2_test.py | 162 key_dtype=dtypes.string,
|
| save_test.py | 339 key_dtype=dtypes.string,
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column.py | [all...] |
| feature_column_v2.py | [all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| lookup_ops_test.py | 709 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64)
749 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
766 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64), oov_buckets)
[all...] |
