Inheritance diagram for tensorflow::serving::TfLiteBatchTask:
Collaboration diagram for tensorflow::serving::TfLiteBatchTask:
Public Member Functions | |
| TfLiteBatchTask (const TfLiteBatchTask &)=delete | |
| TfLiteBatchTask & | operator= (const TfLiteBatchTask &)=delete |
| size_t | size () const override |
| uint64_t | start_time_micros () const |
| void | set_output (Tensor t) |
Static Public Member Functions | |
| static void | CreateTfLiteBatchTask (const std::vector< string > *output_tensor_names, std::vector< Tensor > *outputs, Notification *done, Status *status, std::unique_ptr< TfLiteBatchTask > *batch_task) |
| static void | CreatePartialTfLiteBatchTask (std::vector< int > input_indices, const std::vector< string > *output_tensor_names, std::vector< Tensor > *outputs, std::function< void()> done_callback, ThreadSafeStatus *partial_status, std::unique_ptr< TfLiteBatchTask > *batch_task) |
Public Attributes | |
| Notification * | done |
| Status * | status |
| std::vector< int > | input_indices |
| std::vector< Tensor > | inputs |
| std::vector< Tensor > * | outputs |
| const std::vector< string > * | output_tensor_names |
| RunOptions | run_options |
| const uint64_t | enqueue_time_micros |
| bool | is_partial = false |
| std::function< void()> | done_callback |
| ThreadSafeStatus * | partial_status |
Detailed Description
Definition at line 46 of file tflite_session.h.
The documentation for this class was generated from the following file:
- tensorflow_serving/servables/tensorflow/tflite_session.h
