| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 80 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 …]
|
| D | function_threadpool_test.cc | 72 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 …]
|
| D | process_function_library_runtime_test.cc | 48 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 …]
|
| D | process_function_library_runtime.cc | 48 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 …]
|
| D | function.h | 56 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);
|
| D | function_test.cc | 169 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/ |
| D | partitioned_function_ops.h | 44 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_
|
| D | partitioned_function_ops.cc | 81 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()
|
| D | functional_ops.cc | 32 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 …]
|
| D | function_ops.cc | 242 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/ |
| D | function_handle_cache.h | 27 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_
|
| D | function_handle_cache.cc | 24 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache()
36 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
37 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
39 FunctionLibraryRuntime::Handle h; in Instantiate()
|
| D | function.h | 498 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/ |
| D | cluster_function_library_runtime.h | 39 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,
|
| D | cluster_function_library_runtime_test.cc | 53 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()
|
| D | cluster_function_library_runtime.cc | 30 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/ |
| D | captured_function.h | 81 FunctionLibraryRuntime::DoneCallback done,
96 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle,
103 FunctionLibraryRuntime* const lib_;
104 const FunctionLibraryRuntime::Handle f_handle_;
|
| D | captured_function.cc | 135 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/ |
| D | const_analysis.cc | 35 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()
|
| D | tf2xla_util.h | 151 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
157 FunctionLibraryRuntime::Handle* handle);
166 FunctionLibraryRuntime* flr_;
167 std::map<string, FunctionLibraryRuntime::Handle> handles_;
|
| D | graph_compiler.h | 59 FunctionLibraryRuntime* flib, in GraphCompiler()
83 FunctionLibraryRuntime* flib_;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 42 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/ |
| D | create_xla_launch_op.cc | 68 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()
|
| D | create_xla_launch_op.h | 23 class FunctionLibraryRuntime; variable
30 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 67 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
76 FunctionLibraryRuntime::Options opts; in ComputeAsync()
105 FunctionLibraryRuntime::Handle handle_;
|