Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dsaved_model_bundle_test.cc117 RunOptions run_options; in TEST_F() local
131 RunOptions run_options; in TEST_F() local
143 RunOptions run_options; in TEST_F() local
159 RunOptions run_options; in TEST_F() local
179 RunOptions run_options; in TEST_F() local
195 RunOptions run_options; in TEST_F() local
216 RunOptions run_options; in TEST_F() local
230 RunOptions run_options; in TEST_F() local
242 RunOptions run_options; in TEST_F() local
253 RunOptions run_options; in TEST_F() local
[all …]
H A 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 …]
H A Dloader.cc174 Status RunOnce(const RunOptions& run_options, in RunOnce()
208 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp()
224 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore()
279 const RunOptions& run_options, in LoadSavedModelInternal()
295 const RunOptions& run_options, const string& export_dir, in LoadSavedModel()
354 Status Create(const RunOptions& run_options, const GraphDef& graph) override { in Create()
357 Status Extend(const RunOptions& run_options, const GraphDef& graph) override { in Extend()
360 Status Create(const RunOptions& run_options, GraphDef&& graph) override { in Create()
363 Status Extend(const RunOptions& run_options, GraphDef&& graph) override { in Extend()
366 Status Close(const RunOptions& run_options) override { in Close()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.cc81 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
103 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
118 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
130 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
162 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
173 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
191 const ServiceExecutableRunOptions* run_options) { in ExecuteWrapperBeforeExecution()
279 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
290 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/public/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_runtime.cc261 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal()
301 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
324 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
342 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
365 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
694 const xla::ExecutableRunOptions* run_options, in GetRendezvousKey()
719 const xla::ExecutableRunOptions* run_options, int32_t channel_id_present, in __xla_cpu_runtime_AllToAll()
761 const xla::ExecutableRunOptions* run_options, in __xla_cpu_runtime_AllReduce()
811 const xla::ExecutableRunOptions* run_options, void* output_buffer) { in __xla_cpu_runtime_ReplicaId()
821 const xla::ExecutableRunOptions* run_options, void* output_buffer) { in __xla_cpu_runtime_PartitionId()
[all …]
H A Dcpu_executable.cc154 const ExecutableRunOptions* run_options, in ExecuteComputeFunction()
219 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
319 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
367 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream() member
H A Druntime_conv2d.cc35 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv2DF32() local
57 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv2DF16() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dlocal_client.cc56 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions()
121 ExecutableRunOptions run_options) { in RunHelper()
177 ExecutableRunOptions run_options) { in Run()
190 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in Run()
245 ExecutableRunOptions run_options) { in RunAsync()
285 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
326 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options) { in RunAsync()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc419 Error LaunchFunc::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
514 Error Gemm::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
590 const ServiceExecutableRunOptions* run_options, in operator ()()
965 Error Infeed::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1033 Error Outfeed::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1117 const ServiceExecutableRunOptions* run_options, runtime::FlatMemrefView dst, in operator ()()
1178 Error Memset::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1244 Error Fft::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1311 Error Cholesky::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1390 Error TriangularSolve::run(const ServiceExecutableRunOptions* run_options, in run()
[all …]
H A Dgpu_executable.cc285 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
320 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
353 Status MaybeSyncAndProfile(const ServiceExecutableRunOptions* run_options, in MaybeSyncAndProfile()
556 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
563 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
574 const ServiceExecutableRunOptions* run_options, in ExecuteJitRt()
684 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamImpl()
845 const ServiceExecutableRunOptions* run_options, in ExecuteThunksOrJitRt()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session.cc139 Status GrpcSession::Create(const RunOptions& run_options, in Create()
150 Status GrpcSession::Create(const RunOptions& run_options, GraphDef&& graph) { in Create()
183 Status GrpcSession::Extend(const RunOptions& run_options, in Extend()
194 Status GrpcSession::Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend()
201 const RunOptions& run_options, in RunHelper()
287 Status GrpcSession::Run(const RunOptions& run_options, in Run()
301 RunOptions run_options; in Run() local
345 RunOptions run_options; in PRun() local
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession_ref.cc129 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun()
187 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate()
208 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend()
229 Status RecordClose(Session* session, const RunOptions& run_options) { in RecordClose()
410 Status SessionRef::Run(const RunOptions& run_options, in Run()
432 Status SessionRef::Create(const RunOptions& run_options, in Create()
437 Status SessionRef::Extend(const RunOptions& run_options, in Extend()
489 Status SessionRef::Close(const RunOptions& run_options) { in Close()
H A Dtf_session_helper.cc61 const TF_Buffer* run_options, PyObject* feed_dict, in TF_Run_wrapper_helper()
165 void TF_Run_wrapper(TF_DeprecatedSession* session, const TF_Buffer* run_options, in TF_Run_wrapper()
348 const TF_Buffer* run_options, in TF_SessionRun_wrapper_helper()
435 void TF_SessionRun_wrapper(TF_Session* session, const TF_Buffer* run_options, in TF_SessionRun_wrapper()
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_utils.py21 def add_debug_tensor_watch(run_options, argument
77 def watch_graph(run_options, argument
197 def watch_graph_with_denylists(run_options, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A Dsaved_model_test.cc73 tfrt::SavedModel::RunOptions run_options; in TEST_P() local
118 tfrt::SavedModel::RunOptions run_options; in TEST() local
557 tfrt::SavedModel::RunOptions run_options; in TEST() local
602 tfrt::SavedModel::RunOptions run_options; in TEST() local
652 tfrt::SavedModel::RunOptions run_options; in TEST() local
1000 tfrt::SavedModel::RunOptions run_options; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/training/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_test_base.cc154 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
172 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie()
188 const ExecutableRunOptions& run_options) { in ExecuteLocally()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_with_tracking_alloc_test.cc205 RunOptions run_options; in TestHWAccelerator() local
284 RunOptions run_options; in TEST() local
342 RunOptions run_options; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc82 const GraphExecutionRunOptions& run_options, in SetUpRequestContext()
142 const GraphExecutionRunOptions& run_options, in GraphExecutionRunOnFunction()
343 const RunOptions& run_options, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_test.cc662 RunOptions run_options; in TEST_F() local
683 RunOptions run_options; in TEST_F() local
726 RunOptions run_options; in TEST_F() local
757 RunOptions run_options; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlift_variables_test_pass.h78 Status Run(const tensorflow::RunOptions& run_options, in Run()
88 Status Run(const tensorflow::RunOptions& run_options, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dxla_compilation_cache_test_helper.cc93 RunOptions run_options; in ExecuteWithBatch() local
117 RunOptions run_options; in ExecuteWithBatch() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dfake_session.cc121 const tensorflow::RunOptions& run_options, in Run()
131 const tensorflow::RunOptions& run_options, in Run()

1234