/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_;
|