Namespace tflite::gpu::anonymous_namespace{tflite_gpu_runner.cc}
Namespace List > tflite > gpu > anonymous_namespace{tflite_gpu_runner.cc}
Public Functions
Type | Name |
---|---|
ObjectDef | GetSSBOObjectDef (int channels) |
absl::Status | InitializeShapes (const tflite::FlatBufferModel & flatbuffer, const tflite::OpResolver & op_resolver, std::vector< std::vector< int > > * input_shapes, std::vector< std::vector< int > > * output_shapes) |
void | UpdateShapes (const tflite::Interpreter & interpreter, const std::vector< int > & indices, std::vector< std::vector< int > > * shapes) |
Public Functions Documentation
function GetSSBOObjectDef
ObjectDef tflite::gpu::anonymous_namespace{tflite_gpu_runner.cc}::GetSSBOObjectDef (
int channels
)
function InitializeShapes
absl::Status tflite::gpu::anonymous_namespace{tflite_gpu_runner.cc}::InitializeShapes (
const tflite::FlatBufferModel & flatbuffer,
const tflite::OpResolver & op_resolver,
std::vector< std::vector< int > > * input_shapes,
std::vector< std::vector< int > > * output_shapes
)
function UpdateShapes
void tflite::gpu::anonymous_namespace{tflite_gpu_runner.cc}::UpdateShapes (
const tflite::Interpreter & interpreter,
const std::vector< int > & indices,
std::vector< std::vector< int > > * shapes
)
The documentation for this class was generated from the following file /home/friedel/devel/ILLIXR-plugins/hand_tracking/mediapipe/util/tflite/tflite_gpu_runner.cc