Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/cc/saved_model/
Dsaved_model_bundle_test.cc114 RunOptions run_options; in TEST_F() local
128 RunOptions run_options; in TEST_F() local
140 RunOptions run_options; in TEST_F() local
156 RunOptions run_options; in TEST_F() local
176 RunOptions run_options; in TEST_F() local
192 RunOptions run_options; in TEST_F() local
213 RunOptions run_options; in TEST_F() local
227 RunOptions run_options; in TEST_F() local
239 RunOptions run_options; in TEST_F() local
250 RunOptions run_options; in TEST_F() local
[all …]
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 …]
Dloader.cc151 Status RunOnce(const RunOptions& run_options, in RunOnce()
185 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp()
201 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore()
253 const RunOptions& run_options, in LoadSavedModelInternal()
269 const RunOptions& run_options, const string& export_dir, in LoadSavedModel()
326 Status Create(const RunOptions& run_options, const GraphDef& graph) override { in Create()
329 Status Extend(const RunOptions& run_options, const GraphDef& graph) override { in Extend()
332 Status Create(const RunOptions& run_options, GraphDef&& graph) override { in Create()
335 Status Extend(const RunOptions& run_options, GraphDef&& graph) override { in Extend()
338 Status Close(const RunOptions& run_options) override { in Close()
[all …]
/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/compiler/xla/service/
Dexecutable.cc82 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
104 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
119 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
131 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
163 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
174 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
192 const ServiceExecutableRunOptions* run_options) { in ExecuteWrapperBeforeExecution()
280 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
291 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/python/debug/lib/
Ddebug_utils.py26 def add_debug_tensor_watch(run_options, argument
82 def watch_graph(run_options, argument
202 def watch_graph_with_denylists(run_options, argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc55 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions()
120 ExecutableRunOptions run_options) { in RunHelper()
174 ExecutableRunOptions run_options) { in Run()
187 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in Run()
242 ExecutableRunOptions run_options) { in RunAsync()
282 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
323 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
Dlocal_client.h104 const ExecutableRunOptions& run_options, in AsyncCallAndBlockHostUntilDone()
/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/compiler/xla/service/cpu/
Dcpu_runtime.cc233 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
257 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
276 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
300 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
587 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal()
596 const xla::ExecutableRunOptions* run_options, in GetRendezvousKey()
619 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present, in __xla_cpu_runtime_AllToAll()
660 const xla::ExecutableRunOptions* run_options, in __xla_cpu_runtime_AllReduce()
710 const xla::ExecutableRunOptions* run_options, void* output_buffer) { in __xla_cpu_runtime_ReplicaId()
720 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present, in __xla_cpu_runtime_CollectivePermute()
Dcpu_executable.cc149 const ExecutableRunOptions* run_options, in ExecuteComputeFunction()
226 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
326 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
374 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream() member
Druntime_conv2d.cc38 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF32() local
62 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF16() local
Druntime_fft.cc33 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
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc128 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun()
186 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate()
207 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend()
228 Status RecordClose(Session* session, const RunOptions& run_options) { in RecordClose()
409 Status SessionRef::Run(const RunOptions& run_options, in Run()
431 Status SessionRef::Create(const RunOptions& run_options, in Create()
436 Status SessionRef::Extend(const RunOptions& run_options, in Extend()
488 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/compiler/xla/service/gpu/
Dgpu_executable.cc95 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
129 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
399 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
407 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
417 const ServiceExecutableRunOptions* run_options, VariantArguments arguments, in ExecuteAsyncOnStreamImpl()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.h78 Status Run(const tensorflow::RunOptions& run_options, in Run()
88 Status Run(const tensorflow::RunOptions& run_options, in Run()
Dlift_variables_test_pass.cc75 Status Run(const tensorflow::RunOptions& run_options, in Run()
85 Status Run(const tensorflow::RunOptions& run_options, in Run()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc55 const ServiceExecutableRunOptions* run_options, in Evaluate()
/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.cc204 RunOptions run_options; in TestHWAccelerator() local
282 RunOptions run_options; in TEST() local
339 RunOptions run_options; in TEST() local

123