Home
last modified time | relevance | path

Searched refs:FunctionLibraryRuntime (Results 1 – 25 of 75) sorted by relevance

123

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h80 FunctionLibraryRuntime* GetFLR(const string& device_name) const;
90 FunctionLibraryRuntime::Handle AddHandle(
92 FunctionLibraryRuntime::LocalHandle local_handle);
95 FunctionLibraryRuntime::Handle GetHandle(const string& function_key) const;
101 FunctionLibraryRuntime::LocalHandle GetHandleOnDevice(
102 const string& device_name, FunctionLibraryRuntime::Handle handle) const;
109 Status GetOutputDevices(FunctionLibraryRuntime::Handle handle,
115 FunctionLibraryRuntime::Handle handle) const;
121 const FunctionLibraryRuntime::InstantiateOptions& options,
122 FunctionLibraryRuntime::Handle* handle);
[all …]
Dfunction_threadpool_test.cc72 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
73 FunctionLibraryRuntime::Options opts, in Run()
110 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
112 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
116 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
118 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
119 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
123 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
128 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
133 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
[all …]
Dprocess_function_library_runtime_test.cc48 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
49 FunctionLibraryRuntime::LocalHandle* handle) override { in Instantiate()
56 void Run(const FunctionLibraryRuntime::Options& opts, in Run()
57 FunctionLibraryRuntime::LocalHandle handle, in Run()
59 FunctionLibraryRuntime::DoneCallback done) override {} in Run()
100 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Instantiate()
101 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
154 Status Run(const string& name, FunctionLibraryRuntime::Options opts, in Run()
156 const FunctionLibraryRuntime::InstantiateOptions& instantiate_opts, in Run()
158 FunctionLibraryRuntime::Handle handle; in Run()
[all …]
Dprocess_function_library_runtime.cc48 const FunctionLibraryRuntime::InstantiateOptions& options) { in DistributedInit()
148 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation()
159 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext()
181 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
198 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandle( in AddHandle()
200 FunctionLibraryRuntime::LocalHandle local_handle) { in AddHandle()
205 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandleLocked( in AddHandleLocked()
207 FunctionLibraryRuntime::LocalHandle local_handle) { in AddHandleLocked()
216 FunctionLibraryRuntime::Handle
227 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::GetHandle( in GetHandle()
[all …]
Dfunction.h56 std::unique_ptr<FunctionLibraryRuntime> NewFunctionLibraryRuntime(
66 std::unique_ptr<FunctionLibraryRuntime> NewFunctionLibraryRuntime(
140 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g,
142 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g);
249 bool ExpandInlineFunctions(FunctionLibraryRuntime* lib, Graph* graph,
266 inline bool ExpandInlineFunctions(FunctionLibraryRuntime* lib, Graph* graph) { in ExpandInlineFunctions()
275 FunctionLibraryRuntime& flr,
276 FunctionLibraryRuntime::Handle* handle);
Dfunction_test.cc169 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
170 FunctionLibraryRuntime::Options opts, in Run()
207 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
209 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
213 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
215 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
216 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
220 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
225 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
230 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.h44 Status FillOutputDevices(const FunctionLibraryRuntime& lib,
46 FunctionLibraryRuntime::InstantiateOptions* opts);
48 Status Instantiate(FunctionLibraryRuntime* lib, OpKernelContext* ctx,
50 FunctionLibraryRuntime::Handle* handle);
52 void RunFunction(FunctionLibraryRuntime::Handle handle,
54 FunctionLibraryRuntime* lib, OpKernelContext* ctx,
66 gtl::FlatMap<FunctionLibraryRuntime*, FunctionLibraryRuntime::Handle> handles_
Dpartitioned_function_ops.cc81 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
99 FunctionLibraryRuntime::Handle handle; in ComputeAsync()
129 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
130 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
153 Status PartitionedCallOp::Instantiate(FunctionLibraryRuntime* lib, in Instantiate()
156 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
170 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
210 void PartitionedCallOp::RunFunction(FunctionLibraryRuntime::Handle handle, in RunFunction()
212 FunctionLibraryRuntime* lib, in RunFunction()
214 FunctionLibraryRuntime::Options run_opts; in RunFunction()
Dfunctional_ops.cc32 typedef FunctionLibraryRuntime::Handle FHandle;
38 Status Instantiate(FunctionLibraryRuntime* lib, const NameAttrList& func, in Instantiate()
39 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
115 void SetRunOptions(OpKernelContext* ctx, FunctionLibraryRuntime::Options* opts, in SetRunOptions()
208 FunctionLibraryRuntime* const lib_;
209 FunctionLibraryRuntime::Options opts_;
300 FunctionLibraryRuntime* const lib_;
301 FunctionLibraryRuntime::Options opts_;
381 FunctionLibraryRuntime* const lib_;
382 FunctionLibraryRuntime::Options opts_;
[all …]
Dfunction_ops.cc242 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
247 FunctionLibraryRuntime::Handle handle; in ComputeAsync()
251 FunctionLibraryRuntime::Options opts; in ComputeAsync()
304 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
319 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in ComputeAsync()
324 FunctionLibraryRuntime::Handle handle; in ComputeAsync()
349 FunctionLibraryRuntime::Options opts; in ComputeAsync()
/external/tensorflow/tensorflow/core/framework/
Dfunction_handle_cache.h27 explicit FunctionHandleCache(FunctionLibraryRuntime* lib);
35 FunctionLibraryRuntime::InstantiateOptions options,
36 FunctionLibraryRuntime::Handle* handle);
44 FunctionLibraryRuntime* lib_ = nullptr; // not owned
46 std::unordered_map<string, FunctionLibraryRuntime::Handle> handles_
Dfunction_handle_cache.cc24 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache()
36 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
37 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
39 FunctionLibraryRuntime::Handle h; in Instantiate()
Dfunction.h498 class FunctionLibraryRuntime {
500 virtual ~FunctionLibraryRuntime() {} in ~FunctionLibraryRuntime()
693 FunctionLibraryRuntime** out_flr) = 0;
711 const FunctionLibraryRuntime::InstantiateOptions& options);
716 const FunctionLibraryRuntime::Handle kInvalidHandle = -1;
717 const FunctionLibraryRuntime::LocalHandle kInvalidLocalHandle = -1;
718 typedef std::function<Status(FunctionLibraryRuntime*, const NodeDef&,
731 const FunctionLibraryRuntime::InstantiateOptions& options,
732 FunctionLibraryRuntime::LocalHandle* handle) = 0;
735 virtual void Run(const FunctionLibraryRuntime::Options& opts,
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.h39 const FunctionLibraryRuntime::InstantiateOptions& options,
40 FunctionLibraryRuntime::LocalHandle* handle) override;
42 void Run(const FunctionLibraryRuntime::Options& opts,
43 FunctionLibraryRuntime::LocalHandle handle,
45 FunctionLibraryRuntime::DoneCallback done) override;
50 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g,
Dcluster_function_library_runtime_test.cc53 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g, in ConstructFunctionGraphHelper()
62 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
63 FunctionLibraryRuntime::LocalHandle* local_handle) { in Instantiate()
71 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
73 FunctionLibraryRuntime::LocalHandle handle; in InstantiateAndRun()
78 FunctionLibraryRuntime::Options opts; in InstantiateAndRun()
107 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
312 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
327 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
Dcluster_function_library_runtime.cc30 const FunctionLibraryRuntime::InstantiateOptions& options, GraphDef* g, in ConstructFunctionGraph()
122 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
123 FunctionLibraryRuntime::LocalHandle* handle) { in Instantiate()
167 const FunctionLibraryRuntime::Options& opts, in Run()
168 FunctionLibraryRuntime::LocalHandle handle, gtl::ArraySlice<Tensor> args, in Run()
169 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h81 FunctionLibraryRuntime::DoneCallback done,
96 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle,
103 FunctionLibraryRuntime* const lib_;
104 const FunctionLibraryRuntime::Handle f_handle_;
Dcaptured_function.cc135 FunctionLibraryRuntime* lib = ctx->lib(); in Instantiate()
136 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate()
143 FunctionLibraryRuntime::Handle f_handle; in Instantiate()
284 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle, in InstantiatedCapturedFunction()
303 FunctionLibraryRuntime::Options f_opts; in Run()
339 FunctionLibraryRuntime::Options f_opts; in RunWithBorrowedArgs()
375 FunctionLibraryRuntime::Options f_opts; in RunInstantiated()
411 FunctionLibraryRuntime::DoneCallback done, const string& prefix) const { in RunAsync()
418 FunctionLibraryRuntime::Options f_opts; in RunAsync()
446 const FunctionLibraryRuntime::DoneCallback& done, in RunAsync()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc35 FunctionLibraryRuntime* flib_runtime);
42 FunctionLibraryRuntime* flib_runtime, in BackwardsConstAnalysis()
130 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
142 FunctionLibraryRuntime::Handle cond_handle; in GetCompileTimeConstInputs()
143 FunctionLibraryRuntime::Handle body_handle; in GetCompileTimeConstInputs()
Dtf2xla_util.h151 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
157 FunctionLibraryRuntime::Handle* handle);
166 FunctionLibraryRuntime* flr_;
167 std::map<string, FunctionLibraryRuntime::Handle> handles_;
Dgraph_compiler.h59 FunctionLibraryRuntime* flib, in GraphCompiler()
83 FunctionLibraryRuntime* flib_;
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h42 class FunctionLibraryRuntime; variable
66 FunctionLibraryRuntime* flr, in KernelAndDevice()
123 FunctionLibraryRuntime* const flr_; // can be null
134 FunctionLibraryRuntime* flr, in KernelAndDeviceOp()
194 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
232 FunctionLibraryRuntime::Handle handle_;
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op.cc68 Status CompilationRequested(const FunctionLibraryRuntime& flr, in CompilationRequested()
114 Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, in GetBodyAndConstantsAndResources()
119 FunctionLibraryRuntime::Handle handle; in GetBodyAndConstantsAndResources()
153 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def, in CreateXlaLaunchOp()
Dcreate_xla_launch_op.h23 class FunctionLibraryRuntime; variable
30 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfunction_ops.cc67 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
76 FunctionLibraryRuntime::Options opts; in ComputeAsync()
105 FunctionLibraryRuntime::Handle handle_;

123