| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 114 FunctionLibraryRuntime* GetFLR(const string& device_name) const;
117 Status GetRetTypes(FunctionLibraryRuntime::Handle h,
128 FunctionLibraryRuntime::Handle AddHandle(
130 FunctionLibraryRuntime::LocalHandle local_handle);
133 FunctionLibraryRuntime::Handle GetHandle(const string& function_key) const;
143 FunctionLibraryRuntime::LocalHandle GetHandleOnDevice(
144 const string& device_name, FunctionLibraryRuntime::Handle handle,
152 Status GetOutputDevices(FunctionLibraryRuntime::Handle handle,
158 FunctionLibraryRuntime::Handle handle) const;
164 const FunctionLibraryRuntime::InstantiateOptions& options,
[all …]
|
| D | process_function_library_runtime_test.cc | 57 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
58 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
59 FunctionLibraryRuntime::DoneCallback done) override { in Instantiate()
68 void Run(const FunctionLibraryRuntime::Options& opts, in Run()
69 FunctionLibraryRuntime::LocalHandle handle, in Run()
71 FunctionLibraryRuntime::DoneCallback done) override {} in Run()
73 void Run(const FunctionLibraryRuntime::Options& opts, in Run()
74 FunctionLibraryRuntime::LocalHandle handle, in Run()
76 FunctionLibraryRuntime::DoneCallback done) override {} in Run()
78 void CleanUp(uint64 step_id, FunctionLibraryRuntime::LocalHandle handle, in CleanUp()
[all …]
|
| D | function_threadpool_test.cc | 77 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
78 FunctionLibraryRuntime::Options opts, in Run()
115 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
117 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
121 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
123 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
124 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
128 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
133 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
138 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
[all …]
|
| D | process_function_library_runtime.cc | 66 const FunctionLibraryRuntime::InstantiateOptions& options, in DistributedInit()
67 FunctionLibraryRuntime::DoneCallback done) { in DistributedInit()
100 std::unique_ptr<FunctionLibraryRuntime>>), in ProcessFunctionLibraryRuntime()
155 FunctionLibraryRuntime::Handle h, DataTypeVector* ret_types) { in GetRetTypes()
156 FunctionLibraryRuntime* flr = nullptr; in GetRetTypes()
177 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation()
188 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext()
233 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
251 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandle( in AddHandle()
253 FunctionLibraryRuntime::LocalHandle local_handle) { in AddHandle()
[all …]
|
| D | graph_optimizer.h | 74 void Optimize(FunctionLibraryRuntime* runtime, Env* env, const Device* device,
79 FunctionLibraryRuntime* runtime, Env* env, const Device* device,
105 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g,
107 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g);
|
| D | function_test.cc | 177 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
178 FunctionLibraryRuntime::Options opts, in Run()
215 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
217 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
221 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
223 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
224 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
228 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
233 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
238 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
[all …]
|
| D | graph_optimizer.cc | 38 FunctionLibraryRuntime* runtime, Env* env, const Device* device, in Optimize()
142 void GraphOptimizer::Optimize(FunctionLibraryRuntime* runtime, Env* env, in Optimize()
153 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g, in OptimizeGraph()
164 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g) { in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 42 const FunctionLibraryRuntime::InstantiateOptions& options,
43 FunctionLibraryRuntime::LocalHandle* handle,
44 FunctionLibraryRuntime::DoneCallback done) override;
46 void Run(const FunctionLibraryRuntime::Options& opts,
47 FunctionLibraryRuntime::LocalHandle handle,
49 FunctionLibraryRuntime::DoneCallback done) override;
51 void Run(const FunctionLibraryRuntime::Options& opts,
52 FunctionLibraryRuntime::LocalHandle handle,
54 FunctionLibraryRuntime::DoneCallback done) override;
56 void CleanUp(uint64 step_id, FunctionLibraryRuntime::LocalHandle handle,
[all …]
|
| D | cluster_function_library_runtime_test.cc | 56 const FunctionLibraryRuntime::InstantiateOptions& options, in ConstructFunctionGraphHelper()
66 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
67 FunctionLibraryRuntime::LocalHandle* local_handle, in Instantiate()
68 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
76 const FunctionLibraryRuntime::InstantiateOptions& options, in InstantiateAndRun()
78 FunctionLibraryRuntime::LocalHandle handle; in InstantiateAndRun()
92 FunctionLibraryRuntime::Options opts; in InstantiateAndRun()
125 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
383 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
398 FunctionLibraryRuntime::InstantiateOptions instantiate_opts; in TEST_F()
|
| D | cluster_function_library_runtime.cc | 38 const FunctionLibraryRuntime::InstantiateOptions& options, in ConstructFunctionGraph()
183 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
184 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
185 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
258 const FunctionLibraryRuntime::Options& opts, in Run()
259 FunctionLibraryRuntime::LocalHandle handle, gtl::ArraySlice<Tensor> args, in Run()
260 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
334 const FunctionLibraryRuntime::Options& opts, in Run()
335 FunctionLibraryRuntime::LocalHandle handle, in Run()
337 FunctionLibraryRuntime::DoneCallback done) { in Run()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 52 const FunctionLibraryRuntime::InstantiateOptions& options,
53 FunctionLibraryRuntime::LocalHandle* handle,
54 FunctionLibraryRuntime::DoneCallback done) override;
63 void Run(const FunctionLibraryRuntime::Options& opts,
64 FunctionLibraryRuntime::LocalHandle handle,
66 FunctionLibraryRuntime::DoneCallback done) override;
71 void Run(const FunctionLibraryRuntime::Options& opts,
72 FunctionLibraryRuntime::LocalHandle handle,
74 FunctionLibraryRuntime::DoneCallback done) override;
76 void CleanUp(uint64 step_id, FunctionLibraryRuntime::LocalHandle handle,
[all …]
|
| D | cluster_function_library_runtime.cc | 50 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
51 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
52 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
116 const FunctionLibraryRuntime::Options& opts, in Run()
117 FunctionLibraryRuntime::LocalHandle handle, gtl::ArraySlice<Tensor> args, in Run()
118 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
145 const FunctionLibraryRuntime::Options& opts, in Run()
146 FunctionLibraryRuntime::LocalHandle handle, in Run()
148 FunctionLibraryRuntime::DoneCallback done) { in Run()
256 uint64 step_id, FunctionLibraryRuntime::LocalHandle handle, in CleanUp()
[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,
67 gtl::FlatMap<FunctionLibraryRuntime*, FunctionLibraryRuntime::Handle> handles_
|
| D | partitioned_function_ops.cc | 87 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
105 FunctionLibraryRuntime::Handle handle; in ComputeAsync()
135 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
136 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
168 Status PartitionedCallOp::Instantiate(FunctionLibraryRuntime* lib, in Instantiate()
171 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
172 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
233 void PartitionedCallOp::RunFunction(FunctionLibraryRuntime::Handle handle, in RunFunction()
235 FunctionLibraryRuntime* lib, in RunFunction()
237 FunctionLibraryRuntime::Options run_opts; in RunFunction()
|
| D | batch_kernels.cc | 89 FunctionLibraryRuntime::Handle fhandle, in Create()
90 FunctionLibraryRuntime* flib, in Create()
112 FunctionLibraryRuntime::Handle fhandle, FunctionLibraryRuntime* flib, in Create()
130 BatchResource(FunctionLibraryRuntime::Handle fhandle, in BatchResource()
131 FunctionLibraryRuntime* flib, std::shared_ptr<BatcherT> batcher, in BatchResource()
141 BatchResource(FunctionLibraryRuntime::Handle fhandle, in BatchResource()
142 FunctionLibraryRuntime* flib, in BatchResource()
158 FunctionLibraryRuntime::Options opts; in ProcessFuncBatchImpl()
183 FunctionLibraryRuntime::Handle fhandle_;
184 FunctionLibraryRuntime* flib_;
[all …]
|
| D | functional_ops.cc | 37 typedef FunctionLibraryRuntime::Handle FHandle;
43 Status Instantiate(FunctionLibraryRuntime* lib, const NameAttrList& func, in Instantiate()
44 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
49 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
50 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
113 void SetRunOptions(OpKernelContext* ctx, FunctionLibraryRuntime::Options* opts, in SetRunOptions()
152 std::unordered_map<FunctionLibraryRuntime*, std::pair<FHandle, FHandle>>
200 FunctionLibraryRuntime* const lib_;
201 FunctionLibraryRuntime::Options opts_;
329 FunctionLibraryRuntime* const lib_;
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.h | 157 const Node& node, FunctionLibraryRuntime* lib_runtime,
171 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
176 FunctionLibraryRuntime* lib_runtime) const { in IsCompilableNode()
185 FunctionLibraryRuntime* lib_runtime) { in IsCompilableCall()
204 const Node& node, FunctionLibraryRuntime* lib_runtime,
209 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
213 bool IsCompilableIf(const Node& if_node, FunctionLibraryRuntime* lib_runtime,
218 FunctionLibraryRuntime* lib_runtime,
226 FunctionLibraryRuntime* lib_runtime,
236 FunctionLibraryRuntime* lib_runtime,
[all …]
|
| D | xla_kernel_creator.h | 24 class FunctionLibraryRuntime; variable
33 const FunctionLibraryRuntime& flr,
37 Status CreateKernel(FunctionLibraryRuntime* flr,
|
| /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()
38 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
39 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
41 FunctionLibraryRuntime::Handle h; in Instantiate()
|
| D | function.h | 612 class FunctionLibraryRuntime {
614 virtual ~FunctionLibraryRuntime() {} in ~FunctionLibraryRuntime()
901 FunctionLibraryRuntime** out_flr,
927 const FunctionLibraryRuntime::InstantiateOptions& options);
930 const FunctionLibraryRuntime::Handle kInvalidHandle = -1;
931 const FunctionLibraryRuntime::LocalHandle kInvalidLocalHandle = -1;
940 const FunctionLibraryRuntime& flr,
945 FunctionLibraryRuntime* flr,
974 const FunctionLibraryRuntime::InstantiateOptions& options,
975 FunctionLibraryRuntime::LocalHandle* handle,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 35 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody()
40 FunctionLibraryRuntime::Handle func_handle; in GetFunctionBody()
47 Status GetFunctionBodies(FunctionLibraryRuntime* flib_runtime, in GetFunctionBodies()
53 FunctionLibraryRuntime::Handle func_handle; in GetFunctionBodies()
64 std::vector<int>* const_input_idxs, FunctionLibraryRuntime* flib_runtime) { in CondConstInputIndices()
91 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
174 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
187 FunctionLibraryRuntime* flib_runtime, in BackwardsConstAnalysis()
289 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
|
| D | tf2xla_util.h | 152 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
158 FunctionLibraryRuntime::Handle* handle);
167 FunctionLibraryRuntime* flr_;
168 std::map<string, FunctionLibraryRuntime::Handle> handles_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.h | 215 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle,
269 FunctionLibraryRuntime::DoneCallback done,
274 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle,
283 FunctionLibraryRuntime* const lib_; // Not owned.
284 const FunctionLibraryRuntime::Handle f_handle_;
|
| D | iterator_ops.h | 40 FunctionLibraryRuntime* flr);
80 FunctionLibraryRuntime* flr, in State()
98 FunctionLibraryRuntime* flr() { return flr_; } in flr()
116 FunctionLibraryRuntime* flr_ = nullptr; // not owned
159 FunctionLibraryRuntime* CreatePrivateFLR(
187 FunctionLibraryRuntime* lib,
|