| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 71 static SessionOptions MakeSessionOptions( in MakeSessionOptions()
78 SessionOptions options;
146 SessionOptions opts = MakeSessionOptions("0", 0, 1, {}, {}, {}, in TEST_F()
173 SessionOptions opts = MakeSessionOptions("0", 0, 1, {}, {}, {}, in TEST_F()
206 SessionOptions opts = MakeSessionOptions("0,abc"); in TEST_F()
215 SessionOptions opts = MakeSessionOptions("100"); in TEST_F()
225 SessionOptions opts = MakeSessionOptions("0,0"); in TEST_F()
235 SessionOptions opts = MakeSessionOptions("0", 0.1, 1, {{}}); in TEST_F()
247 SessionOptions opts = MakeSessionOptions("0", 0, 0, {{}}); in TEST_F()
259 SessionOptions opts = MakeSessionOptions("0", 0, 8, {{}, {}}); in TEST_F()
[all …]
|
| D | gpu_device_factory.cc | 31 GPUDevice(const SessionOptions& options, const string& name, in GPUDevice()
65 const SessionOptions& options, const string& name, Bytes memory_limit, in CreateGPUDevice()
83 GPUCompatibleCPUDevice(const SessionOptions& options, const string& name, in GPUCompatibleCPUDevice()
119 Status CreateDevices(const SessionOptions& options, const string& name_prefix, in CreateDevices()
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | saved_model_bundle_test.cc | 116 SessionOptions session_options; in TEST_F()
130 SessionOptions session_options; in TEST_F()
142 SessionOptions session_options; in TEST_F()
158 SessionOptions session_options; in TEST_F()
180 SessionOptions session_options; in TEST_F()
196 SessionOptions session_options; in TEST_F()
213 SessionOptions session_options; in TEST_F()
229 SessionOptions session_options; in TEST_F()
241 SessionOptions session_options; in TEST_F()
254 SessionOptions session_options; in TEST_F()
[all …]
|
| D | saved_model_bundle_lite_test.cc | 114 SessionOptions session_options; in TEST_F()
128 SessionOptions session_options; in TEST_F()
141 SessionOptions session_options; in TEST_F()
154 SessionOptions session_options; in TEST_F()
170 SessionOptions session_options; in TEST_F()
187 SessionOptions session_options; in TEST_F()
203 SessionOptions session_options; in TEST_F()
215 SessionOptions session_options; in TEST_F()
228 SessionOptions session_options; in TEST_F()
256 SessionOptions session_options; in TEST_F()
|
| D | loader.h | 111 Status LoadMetagraphIntoSession(const SessionOptions& session_options,
121 Status LoadSavedModel(const SessionOptions& session_options,
133 Status LoadSavedModel(const SessionOptions& session_options,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | session_factory.h | 29 struct SessionOptions;
36 virtual Status NewSession(const SessionOptions& options,
39 virtual bool AcceptsOptions(const SessionOptions& options) = 0;
63 virtual Status Reset(const SessionOptions& options, in Reset()
70 static Status GetFactory(const SessionOptions& options,
|
| D | session_test.cc | 27 SessionOptions options; in TEST()
46 bool AcceptsOptions(const SessionOptions& options) override { in AcceptsOptions()
50 Status NewSession(const SessionOptions& options, in NewSession()
66 SessionOptions options; in TEST()
|
| D | process_util.h | 31 thread::ThreadPool* ComputePool(const SessionOptions& options);
45 int32 NumInterOpThreadsFromSessionOptions(const SessionOptions& options);
49 const SessionOptions& options);
|
| D | process_util.cc | 74 static thread::ThreadPool* InitComputePool(const SessionOptions& options) { in InitComputePool()
88 thread::ThreadPool* ComputePool(const SessionOptions& options) { in ComputePool()
125 int32 NumInterOpThreadsFromSessionOptions(const SessionOptions& options) { in NumInterOpThreadsFromSessionOptions()
157 const SessionOptions& options) { in NewThreadPoolFromSessionOptions()
|
| D | session_options.cc | 22 SessionOptions::SessionOptions() : env(Env::Default()) {} in SessionOptions() function in tensorflow::SessionOptions
|
| D | session.cc | 62 Session* NewSession(const SessionOptions& options) { in NewSession()
76 Status NewSession(const SessionOptions& options, Session** out_session) { in NewSession()
96 Status Reset(const SessionOptions& options, in Reset()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_test.cc | 52 void InitContext(const SessionOptions& opts, in InitContext()
81 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
116 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
144 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
165 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
202 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
242 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT, /*async=*/true); in TEST_F()
309 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT, /*async=*/true); in TEST_F()
342 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
390 InitContext(SessionOptions(), DEVICE_PLACEMENT_EXPLICIT); in TEST_F()
[all …]
|
| D | placement_test.cc | 82 void InitContext(const SessionOptions& opts, in InitContext()
98 SessionOptions opts; in InitDeviceManager()
113 SessionOptions options; in TEST_F()
154 SessionOptions options; in TEST_F()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.h | 38 const SessionOptions &session_options,
41 FallbackState(const SessionOptions &session_options,
50 const SessionOptions &session_options() const { return session_options_; } in session_options()
62 SessionOptions session_options_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_factory.h | 29 struct SessionOptions;
46 static Status AddCpuDevices(const SessionOptions& options,
54 static Status AddDevices(const SessionOptions& options,
62 const SessionOptions& options,
93 const SessionOptions& options, const std::string& name_prefix,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | basic_ops_benchmark_test.cc | 26 static SessionOptions InitOptions() { in InitOptions()
27 SessionOptions opts; in InitOptions()
33 static SessionOptions* GetOptions() { in GetOptions()
34 static SessionOptions opts = InitOptions(); in GetOptions()
|
| D | sdca_ops_test.cc | 29 const SessionOptions* GetSingleThreadedOptions() { in GetSingleThreadedOptions()
30 static const SessionOptions* const kSessionOptions = []() { in GetSingleThreadedOptions()
31 SessionOptions* const result = new SessionOptions(); in GetSingleThreadedOptions()
40 const SessionOptions* GetMultiThreadedOptions() { in GetMultiThreadedOptions()
41 static const SessionOptions* const kSessionOptions = []() { in GetMultiThreadedOptions()
42 SessionOptions* const result = new SessionOptions(); in GetMultiThreadedOptions()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 69 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2D()
78 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2D()
120 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsDepthwiseConv2dNative()
129 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsDepthwiseConv2dNative()
179 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2DShared()
189 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2DShared()
228 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsMatMul()
237 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsMatMul()
|
| D | fuse_convolutions_test.cc | 78 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFuseResizePadAndConv()
87 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFuseResizePadAndConv()
128 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFuseResizeAndConv()
137 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFuseResizeAndConv()
180 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFusePadAndConv()
189 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFusePadAndConv()
|
| D | fold_old_batch_norms_test.cc | 103 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldOldBatchNorms()
112 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldOldBatchNorms()
181 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
190 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
258 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNorms()
267 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNorms()
335 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
344 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
436 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsWithConcat()
445 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsWithConcat()
[all …]
|
| /external/pigweed/pw_tls_client/public/pw_tls_client/ |
| D | options.h | 26 class SessionOptions {
36 constexpr SessionOptions& set_server_name(std::string_view server_name) { in set_server_name()
45 constexpr SessionOptions& set_transport(stream::ReaderWriter& transport) { in set_transport()
|
| /external/pigweed/pw_tls_client_boringssl/ |
| D | tls_client_boringssl.cc | 21 SessionImplementation::SessionImplementation(SessionOptions) { in SessionImplementation() argument
29 Session::Session(const SessionOptions& options) : session_impl_(options) { in Session()
35 Result<Session*> Session::Create(const SessionOptions&) { in Create() argument
|
| /external/pigweed/pw_tls_client_mbedtls/ |
| D | tls_client_mbedtls_test.cc | 22 auto options = SessionOptions().set_transport(stream::NullStream::Instance()); in TEST()
29 auto options = SessionOptions(); in TEST()
36 auto options = SessionOptions().set_transport(stream::NullStream::Instance()); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session.h | 51 explicit GrpcSession(const SessionOptions& options);
54 static Status Create(const SessionOptions& options,
57 static Status Reset(const SessionOptions& options,
118 const SessionOptions options_;
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session_options.h | 28 struct SessionOptions { struct
60 SessionOptions(); argument
|