| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 68 static SessionOptions MakeSessionOptions( in MakeSessionOptions()
72 SessionOptions options;
91 SessionOptions opts = MakeSessionOptions("0,abc"); in TEST_F()
100 SessionOptions opts = MakeSessionOptions("100"); in TEST_F()
110 SessionOptions opts = MakeSessionOptions("0,0"); in TEST_F()
120 SessionOptions opts = MakeSessionOptions("0", 0.1, 1, {{}}); in TEST_F()
132 SessionOptions opts = MakeSessionOptions("0", 0, 0, {{}}); in TEST_F()
144 SessionOptions opts = MakeSessionOptions("0", 0, 8, {{}, {}}); in TEST_F()
158 SessionOptions opts = MakeSessionOptions("0,1", 0, 8, {{}}); in TEST_F()
172 SessionOptions opts = MakeSessionOptions("0"); in TEST_F()
[all …]
|
| D | gpu_device_factory.cc | 30 GPUDevice(const SessionOptions& options, const string& name, in GPUDevice()
64 const SessionOptions& options, const string& name, Bytes memory_limit, in CreateGPUDevice()
82 GPUCompatibleCPUDevice(const SessionOptions& options, const string& name, in GPUCompatibleCPUDevice()
112 Status CreateDevices(const SessionOptions& options, const string& name_prefix, in CreateDevices()
|
| D | gpu_device.h | 54 BaseGPUDevice(const SessionOptions& options, const string& name,
63 Status Init(const SessionOptions& options);
261 Status CreateDevices(const SessionOptions& options, const string& name_prefix,
300 Status CreateGPUDevice(const SessionOptions& options,
306 const SessionOptions& options, const string& name, Bytes memory_limit,
|
| /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 | device_factory.h | 28 struct SessionOptions;
41 static Status AddDevices(const SessionOptions& options,
49 const SessionOptions& options,
54 const SessionOptions& options, const string& name_prefix,
|
| 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 | 48 static thread::ThreadPool* InitComputePool(const SessionOptions& options) { in InitComputePool()
60 thread::ThreadPool* ComputePool(const SessionOptions& options) { in ComputePool()
77 int32 NumInterOpThreadsFromSessionOptions(const SessionOptions& options) { in NumInterOpThreadsFromSessionOptions()
103 const SessionOptions& options) { in NewThreadPoolFromSessionOptions()
|
| D | session_options.cc | 22 SessionOptions::SessionOptions() : env(Env::Default()) {} in SessionOptions() function in tensorflow::SessionOptions
|
| D | session.cc | 54 Session* NewSession(const SessionOptions& options) { in NewSession()
70 Status NewSession(const SessionOptions& options, Session** out_session) { in NewSession()
85 Status Reset(const SessionOptions& options, in Reset()
|
| D | process_util_test.cc | 23 SessionOptions opts; in TEST()
29 SessionOptions opts; in TEST()
|
| D | graph_execution_state.h | 35 struct SessionOptions;
43 const SessionOptions* session_options = nullptr;
196 const SessionOptions* session_options_; // Not owned
|
| D | mkl_threadpool_device_test.cc | 30 SessionOptions options; in TEST()
41 SessionOptions options; in TEST()
|
| D | local_device.h | 28 struct SessionOptions;
36 LocalDevice(const SessionOptions& options,
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader_test.cc | 105 SessionOptions session_options; in TEST_F()
119 SessionOptions session_options; in TEST_F()
132 SessionOptions session_options; in TEST_F()
148 SessionOptions session_options; in TEST_F()
165 SessionOptions session_options; in TEST_F()
181 SessionOptions session_options; in TEST_F()
193 SessionOptions session_options; in TEST_F()
206 SessionOptions session_options; in TEST_F()
234 SessionOptions session_options; in TEST_F()
|
| /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()
435 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsWithConcat()
444 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsWithConcat()
[all …]
|
| /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,
114 const SessionOptions options_;
|
| /external/tensorflow/tensorflow/cc/client/ |
| D | client_session.cc | 37 static SessionOptions MakeDefaultSessionOptions(const string& target);
53 const SessionOptions& session_options) { in ClientSession()
66 SessionOptions ClientSession::Impl::MakeDefaultSessionOptions( in MakeDefaultSessionOptions()
68 SessionOptions options; in MakeDefaultSessionOptions()
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session_options.h | 28 struct SessionOptions { struct
60 SessionOptions(); argument
|
| D | session.h | 241 Status NewSession(const SessionOptions& options, Session** out_session);
268 Status Reset(const SessionOptions& options,
278 Session* NewSession(const SessionOptions& options);
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 45 SessionOptions Devices(int num_cpus, int num_gpus) { in Devices()
46 SessionOptions result; in Devices()
80 SessionOptions Options(const string& target, int placement_period) { in Options()
81 SessionOptions options; in Options()
95 std::unique_ptr<Session> NewRemote(const SessionOptions& options) { in NewRemote()
|