TensorFlow Serving C++ API Documentation
Public Member Functions | Static Public Member Functions | List of all members
tensorflow::serving::internal::TfLiteInterpreterWrapper Class Reference

Public Member Functions

 TfLiteInterpreterWrapper (std::unique_ptr< tflite::ExternalCpuBackendContext > external_context, std::unique_ptr< tflite::Interpreter > interpreter)
 
 TfLiteInterpreterWrapper (std::unique_ptr< tflite::Interpreter > interpreter)
 
tflite::Interpreter * Get ()
 
int GetBatchSize ()
 
void SetBatchSize (int batch_size)
 
TfLiteStatus Invoke ()
 
tensorflow::Status SetStringData (const std::vector< const Tensor * > &tensors, TfLiteTensor *tflite_tensor, int tensor_index, int batch_size)
 

Static Public Member Functions

static Status CreateTfLiteInterpreterWrapper (const tflite::FlatBufferModel &model, const tensorflow::SessionOptions &options, std::unique_ptr< TfLiteInterpreterWrapper > &wrapper)
 

Detailed Description

Definition at line 47 of file tflite_interpreter_pool.h.


The documentation for this class was generated from the following files: