Home
last modified time | relevance | path

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

123456

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.h115 FunctionLibraryRuntime* GetFLR(const string& device_name) const;
118 Status GetRetTypes(FunctionLibraryRuntime::Handle h,
129 FunctionLibraryRuntime::Handle AddHandle(
131 FunctionLibraryRuntime::LocalHandle local_handle);
134 FunctionLibraryRuntime::Handle GetHandle(const string& function_key) const;
144 FunctionLibraryRuntime::LocalHandle GetHandleOnDevice(
145 const string& device_name, FunctionLibraryRuntime::Handle handle,
153 Status GetOutputDevices(FunctionLibraryRuntime::Handle handle,
160 const FunctionLibraryRuntime::InstantiateOptions& options,
161 FunctionLibraryRuntime::Handle* handle);
[all …]
H A Dprocess_function_library_runtime_test.cc59 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
60 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
61 FunctionLibraryRuntime::DoneCallback done) override { in Instantiate()
70 void Run(const FunctionLibraryRuntime::Options& opts, in Run()
71 FunctionLibraryRuntime::LocalHandle handle, in Run()
73 FunctionLibraryRuntime::DoneCallback done) override {} in Run()
75 void Run(const FunctionLibraryRuntime::Options& opts, in Run()
76 FunctionLibraryRuntime::LocalHandle handle, in Run()
78 FunctionLibraryRuntime::DoneCallback done) override {} in Run()
80 void CleanUp(uint64 step_id, FunctionLibraryRuntime::LocalHandle handle, in CleanUp()
[all …]
H A Dprocess_function_library_runtime.cc69 const FunctionLibraryRuntime::InstantiateOptions& options, in DistributedInit()
70 FunctionLibraryRuntime::DoneCallback done) { in DistributedInit()
103 std::unique_ptr<FunctionLibraryRuntime>>), in ProcessFunctionLibraryRuntime()
158 FunctionLibraryRuntime::Handle h, DataTypeVector* ret_types) { in GetRetTypes()
159 FunctionLibraryRuntime* flr = nullptr; in GetRetTypes()
180 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation()
191 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext()
252 FunctionLibraryRuntime* ProcessFunctionLibraryRuntime::GetFLR( in GetFLR()
270 FunctionLibraryRuntime::Handle ProcessFunctionLibraryRuntime::AddHandle( in AddHandle()
272 FunctionLibraryRuntime::LocalHandle local_handle) { in AddHandle()
[all …]
H A Dfunction_threadpool_test.cc77 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 …]
H A Dfunction_test.cc184 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run()
185 FunctionLibraryRuntime::Options opts, in Run()
210 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
212 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
216 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate()
218 const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
219 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
223 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
228 FunctionLibraryRuntime::InstantiateOptions(), args, in InstantiateAndRun()
233 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun()
[all …]
H A Dfunction.cc150 class FunctionLibraryRuntimeOverlay : public FunctionLibraryRuntime {
152 FunctionLibraryRuntimeOverlay(FunctionLibraryRuntime* base_flr, in FunctionLibraryRuntimeOverlay()
199 FunctionLibraryRuntime** out_flr,
203 FunctionLibraryRuntime* base_flr_; // not owned
318 FunctionLibraryRuntime** out_flr, bool skip_flib_def) { in Clone()
325 class FunctionLibraryRuntimeImpl : public FunctionLibraryRuntime {
383 FunctionLibraryRuntime** out_flr,
438 FunctionLibraryRuntime* flr, OpKernel** kernel);
459 void ExecutorArgsFromOptions(const FunctionLibraryRuntime::Options& run_opts,
526 CallOp(FunctionLibraryRuntime::Handle handle, OpKernelConstruction* ctx) in CallOp()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcluster_function_library_runtime.h42 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 …]
H A Dcluster_function_library_runtime.cc38 const FunctionLibraryRuntime::InstantiateOptions& options, in ConstructFunctionGraph()
182 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
183 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
184 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
257 const FunctionLibraryRuntime::Options& opts, in Run()
258 FunctionLibraryRuntime::LocalHandle handle, gtl::ArraySlice<Tensor> args, in Run()
259 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
333 const FunctionLibraryRuntime::Options& opts, in Run()
334 FunctionLibraryRuntime::LocalHandle handle, in Run()
336 FunctionLibraryRuntime::DoneCallback done) { in Run()
[all …]
H A Dcluster_function_library_runtime_test.cc56 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dcluster_function_library_runtime.h54 const FunctionLibraryRuntime::InstantiateOptions& options,
55 FunctionLibraryRuntime::LocalHandle* handle,
56 FunctionLibraryRuntime::DoneCallback done) override;
65 void Run(const FunctionLibraryRuntime::Options& opts,
66 FunctionLibraryRuntime::LocalHandle handle,
68 FunctionLibraryRuntime::DoneCallback done) override;
73 void Run(const FunctionLibraryRuntime::Options& opts,
74 FunctionLibraryRuntime::LocalHandle handle,
76 FunctionLibraryRuntime::DoneCallback done) override;
78 void CleanUp(uint64 step_id, FunctionLibraryRuntime::LocalHandle handle,
[all …]
H A Dcluster_function_library_runtime.cc49 AttrSlice attrs, const FunctionLibraryRuntime::InstantiateOptions& options, in Instantiate()
50 FunctionLibraryRuntime::LocalHandle* handle, in Instantiate()
51 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
115 const FunctionLibraryRuntime::Options& opts, in Run()
116 FunctionLibraryRuntime::LocalHandle handle, gtl::ArraySlice<Tensor> args, in Run()
117 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
144 const FunctionLibraryRuntime::Options& opts, in Run()
145 FunctionLibraryRuntime::LocalHandle handle, in Run()
147 FunctionLibraryRuntime::DoneCallback done) { in Run()
257 uint64 step_id, FunctionLibraryRuntime::LocalHandle handle, in CleanUp()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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,
67 gtl::FlatMap<FunctionLibraryRuntime*, FunctionLibraryRuntime::Handle> handles_
H A Dpartitioned_function_ops.cc88 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
106 FunctionLibraryRuntime::Handle handle; in ComputeAsync()
136 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
137 AttrSlice attrs, FunctionLibraryRuntime::InstantiateOptions* opts) { in FillOutputDevices()
169 Status PartitionedCallOp::Instantiate(FunctionLibraryRuntime* lib, in Instantiate()
172 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
173 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
234 void PartitionedCallOp::RunFunction(FunctionLibraryRuntime::Handle handle, in RunFunction()
236 FunctionLibraryRuntime* lib, in RunFunction()
238 FunctionLibraryRuntime::Options run_opts; in RunFunction()
H A Dfunctional_ops.cc35 typedef FunctionLibraryRuntime::Handle FHandle;
41 Status Instantiate(FunctionLibraryRuntime* lib, const NameAttrList& func, in Instantiate()
42 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
47 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
48 FunctionLibraryRuntime::InstantiateOptions opts; in Instantiate()
111 void SetRunOptions(OpKernelContext* ctx, FunctionLibraryRuntime::Options* opts, in SetRunOptions()
150 std::unordered_map<FunctionLibraryRuntime*, std::pair<FHandle, FHandle>>
198 FunctionLibraryRuntime* const lib_;
199 FunctionLibraryRuntime::Options opts_;
327 FunctionLibraryRuntime* const lib_;
[all …]
H A Dbatch_kernels.cc138 FunctionLibraryRuntime::Handle fhandle, in Create()
139 FunctionLibraryRuntime* flib, in Create()
162 FunctionLibraryRuntime::Handle fhandle, FunctionLibraryRuntime* flib, in Create()
180 BatchResource(FunctionLibraryRuntime::Handle fhandle, in BatchResource()
181 FunctionLibraryRuntime* flib, std::shared_ptr<BatcherT> batcher, in BatchResource()
191 BatchResource(FunctionLibraryRuntime::Handle fhandle, in BatchResource()
192 FunctionLibraryRuntime* flib, in BatchResource()
208 FunctionLibraryRuntime::Options opts; in ProcessFuncBatchImpl()
233 FunctionLibraryRuntime::Handle fhandle_;
234 FunctionLibraryRuntime* flib_;
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.h634 class FunctionLibraryRuntime {
636 virtual ~FunctionLibraryRuntime() {} in ~FunctionLibraryRuntime()
963 FunctionLibraryRuntime** out_flr,
988 const FunctionLibraryRuntime::InstantiateOptions& options);
991 const FunctionLibraryRuntime::Handle kInvalidHandle = -1;
992 const FunctionLibraryRuntime::LocalHandle kInvalidLocalHandle = -1;
1001 const FunctionLibraryRuntime& flr,
1006 FunctionLibraryRuntime* flr,
1035 const FunctionLibraryRuntime::InstantiateOptions& options,
1036 FunctionLibraryRuntime::LocalHandle* handle,
[all …]
H A Dfunction_handle_cache.h27 explicit FunctionHandleCache(FunctionLibraryRuntime* lib);
38 FunctionLibraryRuntime::InstantiateOptions options,
39 FunctionLibraryRuntime::Handle* handle);
47 FunctionLibraryRuntime* lib_ = nullptr; // not owned
49 std::unordered_map<string, FunctionLibraryRuntime::Handle> handles_
H A Dfunction_handle_cache.cc23 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache()
37 FunctionLibraryRuntime::InstantiateOptions options, in Instantiate()
38 FunctionLibraryRuntime::Handle* handle) { in Instantiate()
40 FunctionLibraryRuntime::Handle h; in Instantiate()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dcompilability_check_util.h176 const Node& node, FunctionLibraryRuntime* lib_runtime,
181 FunctionLibraryRuntime* lib_runtime) const { in IsCompilableNode()
200 const Node& node, FunctionLibraryRuntime* lib_runtime,
205 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
209 bool IsCompilableIf(const Node& if_node, FunctionLibraryRuntime* lib_runtime,
214 FunctionLibraryRuntime* lib_runtime,
222 FunctionLibraryRuntime* lib_runtime,
232 FunctionLibraryRuntime* lib_runtime,
288 Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr,
H A Dcompilability_check_util.cc151 const Node& node, FunctionLibraryRuntime* lib_runtime, in FindUncompilableNodes()
214 const Node& if_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableIf()
233 const Node& case_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableCase()
260 const Node& while_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableWhile()
282 FunctionLibraryRuntime* lib_runtime, in ExtractNodeDefAndCheckCompilability()
310 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime, in IsCompilableCall()
324 FunctionLibraryRuntime::Handle handle; in IsCompilableCall()
379 const Node& node, FunctionLibraryRuntime* lib_runtime, in IsCompilableNode()
631 Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, in GetBodyAndConstantsAndResources()
636 FunctionLibraryRuntime::Handle handle; in GetBodyAndConstantsAndResources()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dconst_analysis.cc36 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody()
41 FunctionLibraryRuntime::Handle func_handle; in GetFunctionBody()
48 Status GetFunctionBodies(FunctionLibraryRuntime* flib_runtime, in GetFunctionBodies()
54 FunctionLibraryRuntime::Handle func_handle; in GetFunctionBodies()
65 std::vector<int>* const_input_idxs, FunctionLibraryRuntime* flib_runtime) { in CondConstInputIndices()
92 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
176 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
189 FunctionLibraryRuntime* flib_runtime, in BackwardsConstAnalysis()
298 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.h143 FunctionLibraryRuntime* flr;
217 Status IsMultiDevice(FunctionLibraryRuntime* flr,
292 FunctionLibraryRuntime::DoneCallback done,
299 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle,
308 FunctionLibraryRuntime* const lib_; // Not owned.
309 const FunctionLibraryRuntime::Handle f_handle_;
H A Dcaptured_function.cc174 FunctionLibraryRuntime::Handle fn_handle; in CreateShortCircuitInfo()
540 FunctionLibraryRuntime* lib = params.flr; in Instantiate()
541 FunctionLibraryRuntime::InstantiateOptions inst_opts; in Instantiate()
660 FunctionLibraryRuntime::Handle f_handle; in Instantiate()
696 Status CapturedFunction::IsMultiDevice(FunctionLibraryRuntime* flr, in IsMultiDevice()
769 FunctionLibraryRuntime* lib, FunctionLibraryRuntime::Handle f_handle, in InstantiatedCapturedFunction()
793 FunctionLibraryRuntime::Options f_opts; in Run()
856 FunctionLibraryRuntime::Options f_opts; in RunWithBorrowedArgs()
911 FunctionLibraryRuntime::Options f_opts; in RunInstantiated()
937 FunctionLibraryRuntime::DoneCallback done, in RunAsync()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.h54 class FunctionLibraryRuntime; variable
119 FunctionLibraryRuntime* flr, in KernelAndDevice()
190 FunctionLibraryRuntime* const flr_; // can be null
202 FunctionLibraryRuntime* flr, in KernelAndDeviceOp()
271 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc()
348 std::shared_ptr<FunctionLibraryRuntime::Options> PrepareForRun(
356 FunctionLibraryRuntime::Handle handle_;
/aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/
H A Drpc_ops.cc160 FunctionLibraryRuntime::Handle handle;
161 FunctionLibraryRuntime* lib;
181 FunctionLibraryRuntime* lib, in Register()
182 FunctionLibraryRuntime::Handle fn_handle, in Register()
235 FunctionLibraryRuntime::Options opts; in Call()
239 FunctionLibraryRuntime::Handle handle = fn_metadata.handle; in Call()
240 FunctionLibraryRuntime* fn_lib = fn_metadata.lib; in Call()
321 FunctionLibraryRuntime* lib, in Register()
322 FunctionLibraryRuntime::Handle fn_handle, in Register()
642 FunctionLibraryRuntime* lib = ctx->function_library(); in Compute()
[all …]

123456