Class tflite::gpu::TFLiteGPURunner

ClassList > tflite > gpu > TFLiteGPURunner

Public Functions

Type	Name
absl::Status	BindSSBOToInputTensor (GLuint ssbo_id, int input_id)
absl::Status	BindSSBOToOutputTensor (GLuint ssbo_id, int output_id)
absl::Status	Build ()
bool	CanGenerateSerializedBinaryCache ()
bool	CanGenerateSerializedModel ()
void	ForceOpenCL ()
void	ForceOpenCLInitFromSerializedModel ()
void	ForceOpenGL ()
absl::StatusOr< int64_t >	GetInputElements (int id)
std::vector< BHWC >	GetInputShapes ()
absl::StatusOr< int64_t >	GetOutputElements (int id)
std::vector< BHWC >	GetOutputShapes ()
absl::StatusOr< std::vector< uint8_t > >	GetSerializedBinaryCache ()
absl::StatusOr< std::vector< uint8_t > >	GetSerializedModel ()
std::vector< std::vector< int > >	GetTFLiteInputShapes ()
std::vector< std::vector< int > >	GetTFLiteOutputShapes ()
absl::Status	InitializeWithModel (const tflite::FlatBufferModel & flatbuffer, const tflite::OpResolver & op_resolver, bool allow_quant_ops=false)
absl::Status	Invoke ()
void	SetSerializedBinaryCache (std::vector< uint8_t > && cache)
void	SetSerializedModel (std::vector< uint8_t > && serialized_model)
	TFLiteGPURunner (const InferenceOptions & options)
int	inputs_size () const
int	outputs_size () const

Public Functions Documentation

function BindSSBOToInputTensor

absl::Status tflite::gpu::TFLiteGPURunner::BindSSBOToInputTensor (
    GLuint ssbo_id,
    int input_id
) 

---

function BindSSBOToOutputTensor

absl::Status tflite::gpu::TFLiteGPURunner::BindSSBOToOutputTensor (
    GLuint ssbo_id,
    int output_id
) 

---

function Build

absl::Status tflite::gpu::TFLiteGPURunner::Build () 

---

function CanGenerateSerializedBinaryCache

inline bool tflite::gpu::TFLiteGPURunner::CanGenerateSerializedBinaryCache () 

---

function CanGenerateSerializedModel

inline bool tflite::gpu::TFLiteGPURunner::CanGenerateSerializedModel () 

---

function ForceOpenCL

inline void tflite::gpu::TFLiteGPURunner::ForceOpenCL () 

---

function ForceOpenCLInitFromSerializedModel

inline void tflite::gpu::TFLiteGPURunner::ForceOpenCLInitFromSerializedModel () 

---

function ForceOpenGL

inline void tflite::gpu::TFLiteGPURunner::ForceOpenGL () 

---

function GetInputElements

absl::StatusOr< int64_t > tflite::gpu::TFLiteGPURunner::GetInputElements (
    int id
) 

---

function GetInputShapes

inline std::vector< BHWC > tflite::gpu::TFLiteGPURunner::GetInputShapes () 

---

function GetOutputElements

absl::StatusOr< int64_t > tflite::gpu::TFLiteGPURunner::GetOutputElements (
    int id
) 

---

function GetOutputShapes

inline std::vector< BHWC > tflite::gpu::TFLiteGPURunner::GetOutputShapes () 

---

function GetSerializedBinaryCache

absl::StatusOr< std::vector< uint8_t > > tflite::gpu::TFLiteGPURunner::GetSerializedBinaryCache () 

---

function GetSerializedModel

absl::StatusOr< std::vector< uint8_t > > tflite::gpu::TFLiteGPURunner::GetSerializedModel () 

---

function GetTFLiteInputShapes

inline std::vector< std::vector< int > > tflite::gpu::TFLiteGPURunner::GetTFLiteInputShapes () 

---

function GetTFLiteOutputShapes

inline std::vector< std::vector< int > > tflite::gpu::TFLiteGPURunner::GetTFLiteOutputShapes () 

---

function InitializeWithModel

absl::Status tflite::gpu::TFLiteGPURunner::InitializeWithModel (
    const tflite::FlatBufferModel & flatbuffer,
    const tflite::OpResolver & op_resolver,
    bool allow_quant_ops=false
) 

---

function Invoke

absl::Status tflite::gpu::TFLiteGPURunner::Invoke () 

---

function SetSerializedBinaryCache

void tflite::gpu::TFLiteGPURunner::SetSerializedBinaryCache (
    std::vector< uint8_t > && cache
) 

---

function SetSerializedModel

void tflite::gpu::TFLiteGPURunner::SetSerializedModel (
    std::vector< uint8_t > && serialized_model
) 

---

function TFLiteGPURunner

inline explicit tflite::gpu::TFLiteGPURunner::TFLiteGPURunner (
    const InferenceOptions & options
) 

---

function inputs_size

inline int tflite::gpu::TFLiteGPURunner::inputs_size () const

---

function outputs_size

inline int tflite::gpu::TFLiteGPURunner::outputs_size () const

---

---

The documentation for this class was generated from the following file /home/friedel/devel/ILLIXR-plugins/hand_tracking/mediapipe/util/tflite/tflite_gpu_runner.h