Home
last modified time | relevance | path

Searched defs:run_options (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc36 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
60 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
75 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
87 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
119 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
137 const ServiceExecutableRunOptions* run_options) { in ExecuteWrapperBeforeExecution()
229 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
240 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
Dservice_executable_run_options.h37 : run_options_(std::move(run_options)), in run_options_()
41 const ExecutableRunOptions& run_options() const { return run_options_; } in run_options() function
/external/tensorflow/tensorflow/core/public/
Dsession.h142 virtual Status Create(const RunOptions& run_options, const GraphDef& graph) { in Create()
147 virtual Status Extend(const RunOptions& run_options, const GraphDef& graph) { in Extend()
153 virtual Status Create(const RunOptions& run_options, GraphDef&& graph) { in Create()
156 virtual Status Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend()
160 virtual Status Close(const RunOptions& run_options) { in Close()
180 virtual Status Run(const RunOptions& run_options, in Run()
/external/tensorflow/tensorflow/cc/saved_model/
Dsaved_model_bundle_lite_test.cc98 RunOptions run_options; in CheckSavedModelBundle() local
115 RunOptions run_options; in TEST_F() local
129 RunOptions run_options; in TEST_F() local
142 RunOptions run_options; in TEST_F() local
153 RunOptions run_options; in TEST_F() local
169 RunOptions run_options; in TEST_F() local
190 RunOptions run_options; in TEST_F() local
204 RunOptions run_options; in TEST_F() local
216 RunOptions run_options; in TEST_F() local
227 RunOptions run_options; in TEST_F() local
[all …]
Dsaved_model_bundle_test.cc107 RunOptions run_options; in TEST_F() local
121 RunOptions run_options; in TEST_F() local
132 RunOptions run_options; in TEST_F() local
148 RunOptions run_options; in TEST_F() local
169 RunOptions run_options; in TEST_F() local
183 RunOptions run_options; in TEST_F() local
195 RunOptions run_options; in TEST_F() local
206 RunOptions run_options; in TEST_F() local
236 RunOptions run_options; in TEST_F() local
248 RunOptions run_options; in TEST_F() local
Dloader.cc113 Status RunOnce(const RunOptions& run_options, in RunOnce()
147 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp()
198 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore()
280 const RunOptions& run_options, in LoadSavedModelInternal()
325 const RunOptions& run_options, const string& export_dir, in LoadSavedModel()
382 Status Create(const RunOptions& run_options, const GraphDef& graph) override { in Create()
385 Status Extend(const RunOptions& run_options, const GraphDef& graph) override { in Extend()
388 Status Create(const RunOptions& run_options, GraphDef&& graph) override { in Create()
391 Status Extend(const RunOptions& run_options, GraphDef&& graph) override { in Extend()
394 Status Close(const RunOptions& run_options) override { in Close()
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_utils.py26 def add_debug_tensor_watch(run_options, argument
82 def watch_graph(run_options, argument
200 def watch_graph_with_blacklists(run_options, argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc171 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
195 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
214 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
238 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
388 const xla::ExecutableRunOptions* run_options, in __xla_cpu_runtime_AllReduce()
446 const xla::ExecutableRunOptions* run_options, void* output_buffer) { in __xla_cpu_runtime_ReplicaId()
Dcpu_executable.cc155 const ExecutableRunOptions* run_options, in ExecuteComputeFunction()
227 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
286 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
341 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream() member
Druntime_fft.cc33 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenFft() local
Druntime_conv2d.cc36 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF32() local
56 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF16() local
Druntime_fork_join.cc74 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_ParallelForkJoin() local
Druntime_matmul_mkl.cc81 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_MKLMatMulF32() local
96 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_MKLMatMulF64() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc55 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions()
120 ExecutableRunOptions run_options) { in RunHelper()
173 ExecutableRunOptions run_options) { in Run()
233 ExecutableRunOptions run_options) { in RunAsync()
275 ExecutableRunOptions run_options) { in RunAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc138 Status GrpcSession::Create(const RunOptions& run_options, in Create()
149 Status GrpcSession::Create(const RunOptions& run_options, GraphDef&& graph) { in Create()
182 Status GrpcSession::Extend(const RunOptions& run_options, in Extend()
193 Status GrpcSession::Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend()
200 const RunOptions& run_options, in RunHelper()
286 Status GrpcSession::Run(const RunOptions& run_options, in Run()
300 RunOptions run_options; in Run() local
344 RunOptions run_options; in PRun() local
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc127 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun()
185 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate()
206 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend()
227 Status RecordClose(Session* session, const RunOptions& run_options) { in RecordClose()
408 Status SessionRef::Run(const RunOptions& run_options, in Run()
430 Status SessionRef::Create(const RunOptions& run_options, in Create()
435 Status SessionRef::Extend(const RunOptions& run_options, in Extend()
487 Status SessionRef::Close(const RunOptions& run_options) { in Close()
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner.cc91 const RunOptions& run_options) { in StartAndCollectCostGraph()
124 const RunOptions& run_options) { in StartAndCollectCostGraph()
217 void QueueRunner::SetRunArgumentsAndCostGraph(const RunOptions& run_options) { in SetRunArgumentsAndCostGraph()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc157 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
175 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie()
191 const ExecutableRunOptions& run_options) { in ExecuteLocally()
Dlocal_client_aot_test.cc43 xla::ExecutableRunOptions run_options; in TEST_F() local
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer_test.cc65 RunOptions run_options; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc106 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
139 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
303 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc213 RunOptions run_options; in TestHWAccelerator() local
291 RunOptions run_options; in TEST() local
348 RunOptions run_options; in TEST() local
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc157 RunOptions run_options; in TEST_F() local
178 RunOptions run_options; in TEST_F() local
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc105 Status ClientSession::Run(const RunOptions& run_options, const FeedType& inputs, in Run()
131 const RunOptions& run_options, const FeedType& inputs, in Run()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc64 const RunOptions& run_options = RunOptions(), in Run()
219 RunOptions run_options; in TEST_F() local

123