Home
last modified time | relevance | path

Searched defs:TfLiteContext (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.h661 typedef struct TfLiteContext { struct
663 size_t tensors_size;
703 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context, argument
714 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor, argument
717 void (*ReportError)(struct TfLiteContext*, const char* msg, ...); argument
722 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, argument
728 struct TfLiteContext*, int node_index, TfLiteNode** node, argument
734 struct TfLiteContext*, struct TfLiteRegistration registration, argument
743 TfLiteExternalContext* (*GetExternalContext)(struct TfLiteContext*, argument
748 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, argument
[all …]
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h709 typedef struct TfLiteContext { struct
711 size_t tensors_size;
751 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context, argument
762 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor, argument
765 void (*ReportError)(struct TfLiteContext*, const char* msg, ...); argument
770 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, argument
776 struct TfLiteContext*, int node_index, TfLiteNode** node, argument
782 struct TfLiteContext*, struct TfLiteRegistration registration, argument
791 TfLiteExternalContext* (*GetExternalContext)(struct TfLiteContext*, argument
796 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_test.cc330 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
387 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
450 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
564 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
731 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
949 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1018 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1215 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1288 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1345 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dmodel_test.cc59 void* dummy_init(TfLiteContext*, const char*, size_t) { return nullptr; } in dummy_init() argument
60 void dummy_free(TfLiteContext*, void*) {} in dummy_free() argument
61 TfLiteStatus dummy_resize(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_resize() argument
62 TfLiteStatus dummy_invoke(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_invoke() argument
710 Prepare = [](TfLiteContext*, TfLiteDelegate* delegate) -> TfLiteStatus { in TEST() argument
H A Dinterpreter_test.cc1385 Prepare = [](TfLiteContext*, TfLiteDelegate* delegate) -> TfLiteStatus { in TEST() argument
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
H A Dsentencepiece_detokenizer_tflite.cc38 void* Initialize(TfLiteContext* /*context*/, const char* /*buffer*/, in Initialize() argument
42 void Free(TfLiteContext* /*context*/, void* /*buffer*/) {} in Free() argument
H A Dsentencepiece_tokenizer_tflite.cc50 void* Initialize(TfLiteContext* /*context*/, const char* /*buffer*/, in Initialize() argument
54 void Free(TfLiteContext* /*context*/, void* /*buffer*/) {} in Free() argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils_test.cc49 context.ResizeTensor = [](struct TfLiteContext*, TfLiteTensor* tensor, in TEST()
50 TfLiteIntArray* new_size) { in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_disabled.cc47 TfLiteStatus StatefulNnApiDelegate::DoPrepare(TfLiteContext* /* context */, in DoPrepare() argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature_test.cc204 TEST(GetOpSignature, TfLiteContext) { in TEST() argument
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc577 void Free(TfLiteContext* /*context*/, void* buffer) { in Free() argument