| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 89 def _check_table_dtypes(table, key_dtype, value_dtype): argument
112 def __init__(self, key_dtype, value_dtype): argument
127 def key_dtype(self): member in LookupInterface
402 def __init__(self, key_dtype, value_dtype): argument
413 def key_dtype(self): member in TableInitializerBase
513 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
644 key_dtype, argument
839 key_dtype=dtypes.string): argument
983 key_dtype=None): argument
1356 key_dtype=dtypes.string, argument
[all …]
|
| D | map_ops.py | 50 def tensor_map_stack_keys(input_handle, key_dtype): argument
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable.cc | 40 TfLiteType key_dtype; member
51 TfLiteType key_dtype, value_dtype; in InitHashtable() local
|
| D | hashtable_ops_test.cc | 637 explicit HashtableOpModel(const char* table_name, TensorType key_dtype, in HashtableOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 139 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
1000 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1035 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1069 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1102 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| D | lookup_util.cc | 79 DataType key_dtype, int64 key_index, DataType value_dtype, in Init()
343 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes()
365 const DataType& key_dtype = table->key_dtype(); in InitializeTableFromTextFile() local
|
| D | lookup_table_op.h | 215 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 97 TfLiteType key_dtype, in CreateHashtableResourceIfNotAvailable()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | lookup_ops.cc | 95 DataType key_dtype; in ValidateTableType() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 356 DataType key_dtype = DT_INT64; in SparsifyGatherInternal() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 412 const DataType key_dtype = DT_INT64; in TEST_F() local
|