Home
last modified time | relevance | path

Searched refs:fallback_op_handler_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/
Dop_handler_selector.cc40 fallback_op_handler_(fallback_op_handler), in EagerOpHandlerSelector()
43 assert(fallback_op_handler_); in EagerOpHandlerSelector()
109 if (!(*op_handler)) *op_handler = fallback_op_handler_; in SelectFromNodeDef()
Dop_handler_selector.h67 OpHandler* fallback_op_handler_; variable
/external/tensorflow/tensorflow/core/tfrt/eager/
Dtfrt_context.h91 OpHandler* GetFallbackOpHandler() { return fallback_op_handler_; } in GetFallbackOpHandler()
106 OpHandler* fallback_op_handler_; variable
Dtfrt_context.cc102 fallback_op_handler_ = runtime_fallback_op_handler.get(); in TfrtContext()
103 corert_->RegisterOpHandler("tf", fallback_op_handler_); in TfrtContext()
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc361 fallback_op_handler_ = CreateOpHandler(); in SelectorTest()
368 corert_.get(), eager_context_, fallback_op_handler_, in SelectorTest()
393 OpHandler* fallback_op_handler_; member in tfrt::tf::__anone7954bb30111::SelectorTest