tflite_support.task.processor.SearchOptions

View source on GitHub

Options for search processor.

  tflite_support 
 . 
 task 
 . 
 processor 
 . 
 SearchOptions 
 ( 
 index_file_name 
 : 
 Optional 
 [ 
 str 
 ] 
 = 
 None 
 , 
 index_file_content 
 : 
 Optional 
 [ 
 bytes 
 ] 
 = 
 None 
 , 
 max_results 
 : 
 Optional 
 [ 
 int 
 ] 
 = 
 5 
 ) 
 

The index file to search into. Mandatory only if the index is not attached to the output tensor metadata as an AssociatedFile with type SCANN_INDEX_FILE. The index file can be specified by one of the following two ways:

(1) file contents loaded in index_file_content . (2) file path in index_file_name .

If more than one field of these fields is provided, they are used in this precedence order.

Methods

create_from_pb2

View source

  @classmethod 
 
  create_from_pb2 
 ( 
 pb2_obj 
 : 
 _SearchOptionsProto 
 ) 
 -> 
 'SearchOptions' 
 

Creates a SearchOptionsProto object from the given protobuf object.

to_pb2

View source

  to_pb2 
 () 
 -> 
 _SearchOptionsProto 
 

Generates a protobuf object to pass to the C++ layer.

__eq__

View source

  __eq__ 
 ( 
 other 
 : 
 Any 
 ) 
 -> 
 bool 
 

Checks if this object is equal to the given object.