Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/experimental/base/public/
Druntime_builder.h32 class RuntimeBuilder {
34 RuntimeBuilder() : options_(TFE_NewContextOptions()) {} in RuntimeBuilder() function
39 RuntimeBuilder& SetUseTFRT(bool use_tfrt);
51 RuntimeBuilder(RuntimeBuilder&&) = default;
52 RuntimeBuilder& operator=(RuntimeBuilder&&) = default;
56 RuntimeBuilder(const RuntimeBuilder&) = delete;
57 RuntimeBuilder& operator=(const RuntimeBuilder&) = delete;
67 inline RuntimeBuilder& RuntimeBuilder::SetUseTFRT(bool use_tfrt) { in SetUseTFRT()
72 inline std::unique_ptr<Runtime> RuntimeBuilder::Build(Status* status) { in Build()
Druntime.h42 friend class RuntimeBuilder;
Dstatus.h58 friend class RuntimeBuilder;
/external/tensorflow/tensorflow/cc/saved_model/experimental/tests/
Dsaved_model_api_test.cc33 using tensorflow::experimental::cc::RuntimeBuilder;
51 RuntimeBuilder builder; in TEST_P()
75 RuntimeBuilder builder; in TEST_P()
/external/tensorflow/tensorflow/cc/experimental/base/tests/
Dtensorhandle_test.cc35 using tensorflow::experimental::cc::RuntimeBuilder;
55 RuntimeBuilder runtime_builder; in TYPED_TEST()
92 RuntimeBuilder runtime_builder; in TYPED_TEST()
143 RuntimeBuilder runtime_builder; in TYPED_TEST()