/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...] |