Home
last modified time | relevance | path

Searched refs:KernelAndDevice (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h99 class KernelAndDevice : public core::RefCounted {
112 KernelAndDevice( in KernelAndDevice() function
124 ~KernelAndDevice() override {} in ~KernelAndDevice()
190 class KernelAndDeviceOp final : public KernelAndDevice {
198 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
255 class KernelAndDeviceFunc : public KernelAndDevice {
272 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
Dexecute_node.h56 const core::RefCountPtr<KernelAndDevice>& kernel);
97 const core::RefCountPtr<KernelAndDevice>& kernel, in ExecuteNode()
141 const core::RefCountPtr<KernelAndDevice>& kernel_;
153 core::RefCountPtr<KernelAndDevice> kernel, in AsyncExecuteNode()
236 core::RefCountPtr<KernelAndDevice> kernel_;
Dexecute.cc184 const core::RefCountPtr<KernelAndDevice>& kernel) { in ValidateInputTypeAndPlacement()
385 core::RefCountPtr<KernelAndDevice>* out_kernel) { in GetOrCreateKernelAndDevice()
466 core::RefCountPtr<KernelAndDevice> kernel = ctx.GetCachedKernel(cache_key); in GetOrCreateKernelAndDevice()
575 const KernelAndDevice& kernel, const int output_num, Device* output_device, in CreateUnshapedOutput()
609 Status AddOrExecuteNode(core::RefCountPtr<KernelAndDevice> kernel, in AddOrExecuteNode()
707 core::RefCountPtr<KernelAndDevice> kernel; in EagerLocalExecute()
989 TensorHandle** retvals, EagerContext* ctx, KernelAndDevice* kernel, in GetKernelOutputs()
1110 const core::RefCountPtr<KernelAndDevice>& kernel, in EagerKernelExecute()
1317 const core::RefCountPtr<KernelAndDevice> kernel, in EagerKernelExecuteAsync()
1376 core::RefCountPtr<KernelAndDevice> kernel; in EagerLocalExecuteAsync()
Dexecute.h53 const core::RefCountPtr<KernelAndDevice>& kernel,
Dcontext.h245 core::RefCountPtr<KernelAndDevice> GetCachedKernel(Fprint128 cache_key);
247 void AddKernelToCache(Fprint128 cache_key, KernelAndDevice* kernel);
624 std::unordered_map<Fprint128, core::RefCountPtr<KernelAndDevice>,
Dexecute_node.cc66 const core::RefCountPtr<KernelAndDevice>& kernel) { in Init()
Dexecute_node_test.cc110 const core::RefCountPtr<KernelAndDevice> kernel( in TEST()
Dcontext.cc840 core::RefCountPtr<KernelAndDevice> EagerContext::GetCachedKernel( in GetCachedKernel()
847 core::RefCountPtr<KernelAndDevice> new_ref(iter->second.get()); in GetCachedKernel()
853 KernelAndDevice* kernel) { in AddKernelToCache()
855 core::RefCountPtr<KernelAndDevice> new_ref(kernel); in AddKernelToCache()
Dkernel_and_device.cc77 std::function<void(std::function<void()>)>* KernelAndDevice::get_runner() in get_runner()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc44 EagerOperation* op, core::RefCountPtr<KernelAndDevice>* kernel) { in CreateUncachedKernelAndDeviceOp()
104 core::RefCountPtr<KernelAndDevice> kernel; in RunLocalSend()
192 core::RefCountPtr<KernelAndDevice> kernel; in RunLocalRecv()