| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | lookup_ops.py | 108 def check_table_dtypes(table, key_dtype, value_dtype): argument
131 def __init__(self, key_dtype, value_dtype): argument
146 def key_dtype(self): member in LookupInterface
479 def __init__(self, key_dtype, value_dtype): argument
490 def key_dtype(self): member in TableInitializerBase
531 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
662 key_dtype, argument
871 key_dtype=dtypes.string): argument
1015 key_dtype=None): argument
1415 key_dtype=dtypes.string, argument
[all …]
|
| H A D | map_ops.py | 46 def tensor_map_stack_keys(input_handle, key_dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/ |
| H A D | simple_hash_table_test.py | 32 def _use_table(self, key_dtype, value_dtype): argument
45 def test_find_insert_find_eager(self, key_dtype, value_dtype): argument
54 def test_find_insert_find_tf_function(self, key_dtype, value_dtype): argument
|
| H A D | simple_hash_table.py | 52 key_dtype, argument
111 def key_dtype(self): member in SimpleHashTable
|
| H A D | simple_hash_table_op.cc | 70 DataType key_dtype; in ValidateTableType() local
|
| H A D | simple_hash_table_kernel.cc | 279 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | lookup_table_op.cc | 138 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
326 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
661 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function in tensorflow::lookup::MutableDenseHashTable
1080 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1122 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1164 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1205 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| H A D | lookup_util.cc | 84 DataType key_dtype, int64_t key_index, DataType value_dtype, in Init()
349 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes()
381 const DataType& key_dtype = table->key_dtype(); in InitializeTableFromTextFile() local
|
| H A D | lookup_ops_test.cc | 54 typedef int32 key_dtype; typedef
|
| H A D | lookup_table_op.h | 286 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); } in key_dtype() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | lookup_ops.py | 108 key_dtype=dtypes.string, argument
196 key_dtype=dtypes.string, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | legalize_hashtables.cc | 63 auto key_dtype = hashtable_op.key_dtype(); in matchAndRewrite() local
140 auto key_dtype = hashtable.key_dtype(); in checkWhetherGraphHasValidStaticLookupTables() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/ |
| H A D | static_hashtable.cc | 99 TfLiteType key_dtype, in CreateHashtableResourceIfNotAvailable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | lookup_ops.cc | 95 DataType key_dtype; in ValidateTableType() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | sparsify_gather.cc | 356 DataType key_dtype = DT_INT64; in SparsifyGatherInternal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | single_machine_test.cc | 412 const DataType key_dtype = DT_INT64; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_ops_test.cc | 630 explicit HashtableOpModel(const int table_id, TensorType key_dtype, in HashtableOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
| H A D | feature_column_v2.py | 3523 def _make_table_from_tfrecord_gzip_file(self, key_dtype, name): argument
3539 def _make_table(self, key_dtype, state_manager): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | builtin_op_data.h | 496 TfLiteType key_dtype; member
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | builtin_op_data.h | 508 TfLiteType key_dtype; member
|