Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc106 RunOptions run_options; in TEST_F() local
120 RunOptions run_options; in TEST_F() local
131 RunOptions run_options; in TEST_F() local
147 RunOptions run_options; in TEST_F() local
168 RunOptions run_options; in TEST_F() local
182 RunOptions run_options; in TEST_F() local
194 RunOptions run_options; in TEST_F() local
205 RunOptions run_options; in TEST_F() local
235 RunOptions run_options; in TEST_F() local
Dloader.cc112 Status RunOnce(const RunOptions& run_options, in RunOnce()
146 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp()
196 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore()
260 const RunOptions& run_options, in LoadSavedModelInternal()
302 const RunOptions& run_options, const string& export_dir, in LoadSavedModel()
/external/tensorflow/tensorflow/core/public/
Dsession.h129 virtual Status Create(const RunOptions& run_options, const GraphDef& graph) { in Create()
134 virtual Status Extend(const RunOptions& run_options, const GraphDef& graph) { in Extend()
139 virtual Status Close(const RunOptions& run_options) { in Close()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc137 const ExecutableRunOptions* run_options, in ExecuteComputeFunction()
209 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
268 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
279 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
290 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamImpl()
327 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStreamImpl() member
Dcpu_runtime.cc109 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
133 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
152 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
176 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
Druntime_conv2d.cc35 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF32() local
54 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF16() local
Druntime_fft.cc32 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenFft() 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/python/debug/lib/
Ddebug_utils.py26 def add_debug_tensor_watch(run_options, argument
82 def watch_graph(run_options, argument
179 def watch_graph_with_blacklists(run_options, argument
/external/tensorflow/tensorflow/compiler/xla/service/
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
Dexecutable.cc33 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
64 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile, in ExecuteOnStreamWrapper()
Dexecutable.h103 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
111 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsession_bundle.cc131 Status RunRestoreOp(const RunOptions& run_options, const StringPiece export_dir, in RunRestoreOp()
148 Status RunInitOp(const RunOptions& run_options, const StringPiece export_dir, in RunInitOp()
160 const SessionOptions& options, const RunOptions& run_options, in LoadSessionBundleFromPathUsingRunOptionsInternal()
238 const RunOptions& run_options, in LoadSessionBundleFromPathUsingRunOptions()
Dbundle_shim_test.cc69 RunOptions run_options; in LoadAndValidateSavedModelBundle() local
524 RunOptions run_options; in TEST() local
574 RunOptions run_options; in TEST() local
588 RunOptions run_options; in TEST() 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()
480 Status SessionRef::Close(const RunOptions& run_options) { in Close()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc49 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
125 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc151 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
169 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie()
185 const ExecutableRunOptions& run_options) { in ExecuteLocally()
Dlocal_client_aot_test.cc43 xla::ExecutableRunOptions run_options; in TEST_F() local
/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/client/
Dlocal_client.cc56 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions()
145 ExecutableRunOptions run_options) { in Run()
179 const ServiceExecutableRunOptions* run_options, in ExecuteAndDump()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc138 Status GrpcSession::Create(const RunOptions& run_options, in Create()
175 Status GrpcSession::Extend(const RunOptions& run_options, in Extend()
183 const RunOptions& run_options, in RunHelper()
269 Status GrpcSession::Run(const RunOptions& run_options, in Run()
283 RunOptions run_options; in Run() local
327 RunOptions run_options; in PRun() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc67 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
237 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
350 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer_test.cc65 RunOptions run_options; in TEST() local
/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

123