Searched refs:LookupInterface (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | lookup_interface.cc | 24 Status LookupInterface::CheckKeyShape(const TensorShape& shape) { in CheckKeyShape()
33 Status LookupInterface::CheckKeyAndValueTypes(const Tensor& keys, in CheckKeyAndValueTypes()
46 Status LookupInterface::CheckKeyAndValueTensorsHelper(const Tensor& keys, in CheckKeyAndValueTensorsHelper()
64 Status LookupInterface::CheckKeyAndValueTensorsForInsert(const Tensor& keys, in CheckKeyAndValueTensorsForInsert()
69 Status LookupInterface::CheckKeyAndValueTensorsForImport(const Tensor& keys, in CheckKeyAndValueTensorsForImport()
74 Status LookupInterface::CheckKeyTensorForRemove(const Tensor& keys) { in CheckKeyTensorForRemove()
82 Status LookupInterface::CheckFindArguments(const Tensor& key, in CheckFindArguments()
|
| D | lookup_interface.h | 34 class LookupInterface : public ResourceBase {
146 virtual ~LookupInterface() = default;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.h | 37 LookupInterface** table);
39 LookupInterface** table);
41 LookupInterface** table);
51 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype,
|
| D | lookup_table_op.h | 68 [ctx, this](lookup::LookupInterface** ret) in Compute()
70 lookup::LookupInterface* container = new Container(ctx, this); in Compute()
83 lookup::LookupInterface* table = nullptr; in Compute()
86 ->template LookupOrCreate<lookup::LookupInterface>( in Compute()
98 h() = MakeResourceHandle<lookup::LookupInterface>( in Compute()
117 ->template Delete<lookup::LookupInterface>(cinfo_.container(), in ~LookupTableOp()
|
| D | lookup_util.cc | 284 LookupInterface** table) { in GetResourceLookupTable()
292 LookupInterface** table) { in GetReferenceLookupTable()
301 LookupInterface** table) { in GetLookupTable()
313 LookupInterface* lookup_table; in GetInitializableLookupTable()
343 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes()
|
| D | lookup_table_op.cc | 48 class MutableHashTableOfScalars final : public LookupInterface {
169 class MutableHashTableOfTensors final : public LookupInterface {
338 class MutableDenseHashTable final : public LookupInterface {
812 Status GetTable(OpKernelContext* ctx, lookup::LookupInterface** table) { in GetTable()
830 lookup::LookupInterface* table; in Compute()
864 lookup::LookupInterface* table; in Compute()
899 lookup::LookupInterface* table; in Compute()
930 lookup::LookupInterface* table; in Compute()
951 lookup::LookupInterface* table; in Compute()
970 lookup::LookupInterface* table; in Compute()
|
| D | initializable_lookup_table.h | 28 class InitializableLookupTable : public LookupInterface {
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 83 LookupInterface* CreateStaticHashtable(TfLiteType key_type, in CreateStaticHashtable()
103 resources->emplace(resource_id, std::unique_ptr<LookupInterface>(hashtable)); in CreateHashtableResourceIfNotAvailable()
106 LookupInterface* GetHashtableResource(ResourceMap* resources, int resource_id) { in GetHashtableResource()
109 return static_cast<LookupInterface*>(it->second.get()); in GetHashtableResource()
|
| D | lookup_interfaces.h | 33 class LookupInterface : public ResourceBase {
59 LookupInterface* GetHashtableResource(ResourceMap* resources, int resource_id);
|
| D | static_hashtable.h | 34 class StaticHashtable : public tflite::resource::LookupInterface {
76 ::tflite::resource::LookupInterface* CreateStaticHashtable(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 4 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| D | tensorflow.lookup.-static-hash-table.pbtxt | 5 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 4 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 5 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| D | tensorflow.lookup.-static-hash-table.pbtxt | 6 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 4 is_instance: "<class \'tensorflow.python.ops.lookup_ops.LookupInterface\'>"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 109 class LookupInterface(trackable.TrackableResource): class
121 super(LookupInterface, self).__init__()
154 class InitializableLookupTableBase(LookupInterface):
929 class IdTableWithHashBuckets(LookupInterface):
1148 class StaticVocabularyTable(LookupInterface):
1724 class MutableHashTable(LookupInterface):
1981 class DenseHashTable(LookupInterface):
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 509 elif isinstance(obj, lookup_ops.LookupInterface):
|