Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/c/
Dcommon.h621 typedef struct TfLiteContext { struct
640 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context, argument
651 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor, argument
654 void (*ReportError)(struct TfLiteContext*, const char* msg, ...); argument
659 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, argument
665 struct TfLiteContext*, int node_index, TfLiteNode** node, argument
671 struct TfLiteContext*, struct TfLiteRegistration registration, argument
680 TfLiteExternalContext* (*GetExternalContext)(struct TfLiteContext*, argument
685 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, argument
701 void* (*AllocatePersistentBuffer)(struct TfLiteContext* ctx, size_t bytes); argument
[all …]
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_detokenizer_tflite.cc38 void* Initialize(TfLiteContext* /*context*/, const char* /*buffer*/, in Initialize() argument
42 void Free(TfLiteContext* /*context*/, void* /*buffer*/) {} in Free() argument
Dsentencepiece_tokenizer_tflite.cc50 void* Initialize(TfLiteContext* /*context*/, const char* /*buffer*/, in Initialize() argument
54 void Free(TfLiteContext* /*context*/, void* /*buffer*/) {} in Free() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc315 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
374 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
437 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
551 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
718 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
938 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1010 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
1212 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
/external/tensorflow/tensorflow/lite/
Dmodel_test.cc58 void* dummy_init(TfLiteContext*, const char*, size_t) { return nullptr; } in dummy_init() argument
59 void dummy_free(TfLiteContext*, void*) {} in dummy_free() argument
60 TfLiteStatus dummy_resize(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_resize() argument
61 TfLiteStatus dummy_invoke(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_invoke() argument
528 Prepare = [](TfLiteContext*, TfLiteDelegate* delegate) -> TfLiteStatus { in TEST() argument
Dinterpreter_test.cc1311 Prepare = [](TfLiteContext*, TfLiteDelegate* delegate) -> TfLiteStatus { in TEST() argument
1778 Prepare = [](TfLiteContext*, TfLiteDelegate* delegate) -> TfLiteStatus { in DummyLazyDelegateProvider() argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_disabled.cc42 TfLiteStatus StatefulNnApiDelegate::DoPrepare(TfLiteContext* /* context */, in DoPrepare() argument
/external/tensorflow/tensorflow/lite/delegates/
Dutils_test.cc48 context.ResizeTensor = [](struct TfLiteContext*, TfLiteTensor* tensor, in TEST()
49 TfLiteIntArray* new_size) { in TEST()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc577 void Free(TfLiteContext* /*context*/, void* buffer) { in Free() argument