Searched refs:LookupInterface (Results 1 – 21 of 21) 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 | 70 [ctx, this](lookup::LookupInterface** ret) in Compute()
72 lookup::LookupInterface* container = new Container(ctx, this); in Compute()
85 lookup::LookupInterface* table = nullptr; in Compute()
88 ->template LookupOrCreate<lookup::LookupInterface>( in Compute()
99 h() = MakeResourceHandle<lookup::LookupInterface>( in Compute()
118 ->template Delete<lookup::LookupInterface>(cinfo_.container(), in ~LookupTableOp()
147 lookup::LookupInterface* table = new Container(ctx, this); in Compute()
161 ResourceHandle::MakeRefCountingHandle<lookup::LookupInterface>( in Compute()
|
| D | lookup_util.cc | 290 LookupInterface** table) { in GetResourceLookupTable()
298 LookupInterface** table) { in GetReferenceLookupTable()
307 LookupInterface** table) { in GetLookupTable()
319 LookupInterface* lookup_table; in GetInitializableLookupTable()
349 Status CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, in CheckTableDataTypes()
|
| D | lookup_table_op.cc | 54 class MutableHashTableOfScalars final : public LookupInterface {
218 class MutableHashTableOfTensors final : public LookupInterface {
432 class MutableDenseHashTable final : public LookupInterface {
892 Status GetTable(OpKernelContext* ctx, lookup::LookupInterface** table) { in GetTable()
910 lookup::LookupInterface* table; in Compute()
944 lookup::LookupInterface* table; in Compute()
979 lookup::LookupInterface* table; in Compute()
1010 lookup::LookupInterface* table; in Compute()
1031 lookup::LookupInterface* table; in Compute()
1050 lookup::LookupInterface* table; in Compute()
|
| D | lookup_ops_test.cc | 87 auto resource_or = output_handle.GetResource<lookup::LookupInterface>(); in TEST_F()
|
| D | initializable_lookup_table.h | 28 class InitializableLookupTable : public LookupInterface {
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 85 LookupInterface* CreateStaticHashtable(TfLiteType key_type, in CreateStaticHashtable()
105 resources->emplace(resource_id, std::unique_ptr<LookupInterface>(hashtable)); in CreateHashtableResourceIfNotAvailable()
108 LookupInterface* GetHashtableResource(ResourceMap* resources, int resource_id) { in GetHashtableResource()
111 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 {
78 ::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.-mutable-hash-table.pbtxt | 4 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.-mutable-hash-table.pbtxt | 4 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 | 128 class LookupInterface(resource.TrackableResource): class
140 super(LookupInterface, self).__init__()
173 class InitializableLookupTableBase(LookupInterface):
961 class IdTableWithHashBuckets(LookupInterface):
1182 class StaticVocabularyTable(LookupInterface):
1788 class MutableHashTable(LookupInterface):
2083 class DenseHashTable(LookupInterface):
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 554 elif isinstance(obj, lookup_ops.LookupInterface):
|