| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 85 def _check_table_dtypes(table, key_dtype, value_dtype): argument
97 if key_dtype.base_dtype != table.key_dtype:
99 (table.key_dtype, key_dtype))
108 def __init__(self, key_dtype, value_dtype): argument
115 self._key_dtype = dtypes.as_dtype(key_dtype)
123 def key_dtype(self): member in LookupInterface
163 super(InitializableLookupTableBase, self).__init__(initializer.key_dtype,
282 key_dtype=self._initializer.key_dtype,
333 def __init__(self, key_dtype, value_dtype): argument
340 self._key_dtype = dtypes.as_dtype(key_dtype)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 76 DataType key_dtype, int64 key_index, DataType value_dtype, in Init() argument
81 key_ = Tensor(key_dtype, TensorShape({})); in Init()
326 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes() argument
328 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) { in CheckTableDataTypes()
330 "Conflicting key/value dtypes ", DataTypeString(key_dtype), "->", in CheckTableDataTypes()
332 DataTypeString(table.key_dtype()), "-", in CheckTableDataTypes()
343 if (key_index == kLineNumber && table->key_dtype() != DT_INT64) { in InitializeTableFromTextFile()
346 DataTypeString(table->key_dtype())); in InitializeTableFromTextFile()
348 const DataType& key_dtype = table->key_dtype(); in InitializeTableFromTextFile() local
350 if (key_index == kWholeLine && !DataTypeIsInteger(key_dtype) && in InitializeTableFromTextFile()
[all …]
|
| D | lookup_table_op.cc | 128 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
266 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
552 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableDenseHashTable
706 key_dtype(), TensorShape({num_buckets_, key_size}), &key_buckets_, in AllocateBuckets()
794 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute()
830 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute()
867 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype()}; in Compute()
940 DataTypeVector expected_inputs = {expected_input_0, table->key_dtype(), in Compute()
966 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
970 .TypeConstraint<key_dtype>("key_dtype") \
[all …]
|
| D | lookup_table_op.h | 38 template <class Container, class key_dtype, class value_dtype>
82 *table, DataTypeToEnum<key_dtype>::v(), in Compute()
207 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function
|
| D | lookup_util.h | 41 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | lookup_interface.cc | 35 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()
|
| D | lookup_interface.h | 95 virtual DataType key_dtype() const = 0;
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sharded_mutable_dense_hashtable.py | 59 key_dtype, argument
67 self._key_dtype = key_dtype
75 key_dtype=key_dtype,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lookup.-key-value-tensor-initializer.pbtxt | 8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
|
| D | tensorflow.lookup.-text-file-initializer.pbtxt | 8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 9 name: "key_dtype"
26 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
|
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 9 name: "key_dtype"
|
| D | tensorflow.lookup.-static-hash-table.pbtxt | 14 name: "key_dtype"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lookup.-key-value-tensor-initializer.pbtxt | 8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
|
| D | tensorflow.lookup.-text-file-initializer.pbtxt | 8 name: "key_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 9 name: "key_dtype"
26 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
|
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 14 name: "key_dtype"
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 398 const DataType key_dtype = DT_INT64; in TEST_F() local
404 SetNodeAttr("key_dtype", key_dtype, hashtable_node); in TEST_F()
411 SetNodeAttr("dtype", key_dtype, keys_node); in TEST_F()
412 Tensor keys(key_dtype, TensorShape{2}); in TEST_F()
430 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
441 SetNodeAttr("dtype", key_dtype, query_node); in TEST_F()
442 Tensor query(key_dtype, TensorShape({})); in TEST_F()
459 SetNodeAttr("Tin", key_dtype, lookup_node); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops.py | 55 key_dtype=dtypes.string, name=name)
332 key_dtype=self._initializer.key_dtype,
|
| D | lookup_ops_test.py | 351 key_dtype=dtypes.int32)
365 key_dtype=dtypes.int64)
1091 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
1139 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
1142 key_dtype=dtypes.int32)
1161 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
1201 None, oov_buckets, key_dtype=dtypes.int32)
1220 None, num_oov_buckets=5, key_dtype=dtypes.float64)
1226 None, num_oov_buckets=5, key_dtype=dtypes.bool)
1397 key_dtype=dtypes.int32)
[all …]
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | math_utils.py | 538 def __init__(self, key_dtype, value_dtype, default_value, *args, **kwargs): argument
541 key_dtype=key_dtype,
575 key_dtype, argument
584 key_dtype=key_dtype,
613 def check_table_dtypes(self, key_dtype, value_dtype): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 357 DataType key_dtype = DT_INT64; in SparsifyGatherInternal() local
362 SetNodeAttr("dtype", key_dtype, &indices_node); in SparsifyGatherInternal()
373 SetNodeAttr("key_dtype", key_dtype, &hashtable_node); in SparsifyGatherInternal()
381 SetNodeAttr("Tkey", key_dtype, &init_table_node); in SparsifyGatherInternal()
389 SetNodeAttr("Tin", key_dtype, &lookup_node); in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | lookup_ops.cc | 102 DataType key_dtype; in ValidateTableResourceHandle() local
103 TF_RETURN_IF_ERROR(c->GetAttr(key_dtype_attr, &key_dtype)); in ValidateTableResourceHandle()
104 if (key_shape_and_type.dtype != key_dtype) { in ValidateTableResourceHandle()
109 DataTypeString(key_dtype)); in ValidateTableResourceHandle()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MutableHashTableOfTensorsV2.pbtxt | 27 name: "key_dtype"
|
| D | api_def_MutableHashTableOfTensors.pbtxt | 25 name: "key_dtype"
|