Home
last modified time | relevance | path

Searched refs:GetInputShape (Results 1 – 6 of 6) sorted by relevance

/external/libtextclassifier/native/utils/tflite/
Dblacklist_base.cc39 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Resize()
56 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Eval()
Dblacklist_base.h69 virtual TfLiteIntArray* GetInputShape(TfLiteContext* context,
Dstring_projection_base.h120 virtual TfLiteIntArray* GetInputShape(TfLiteContext* context,
Dblacklist.cc63 TfLiteIntArray* GetInputShape(TfLiteContext* context, in GetInputShape() function in tflite::ops::custom::libtextclassifier3::blacklist::BlacklistOp
Dstring_projection_base.cc211 TfLiteIntArray* input_shape = op->GetInputShape(context, node); in Resize()
228 TfLiteIntArray* input_shape = op->GetInputShape(context, node); in Eval()
Dstring_projection.cc548 TfLiteIntArray* GetInputShape(TfLiteContext* context, in GetInputShape() function in tflite::ops::custom::libtextclassifier3::string_projection::StringProjectionOp