Searched refs:KernelAndDeviceOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 100 Status KernelAndDeviceOp::Init(const bool log_device_placement, in Init()
242 Status KernelAndDeviceOp::Run( in Run()
401 tensorflow::Device* KernelAndDeviceOp::OutputDevice(int idx) const { in OutputDevice()
415 tensorflow::Device* KernelAndDeviceOp::OutputResourceDevice(int idx) const { in OutputResourceDevice()
429 Device* KernelAndDeviceOp::InputDevice(int i) const { in InputDevice()
|
| D | kernel_and_device_test.cc | 116 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceInit()
137 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceRun()
|
| D | kernel_and_device.h | 190 class KernelAndDeviceOp final : public KernelAndDevice {
192 KernelAndDeviceOp( in KernelAndDeviceOp() function
203 ~KernelAndDeviceOp() override {} in ~KernelAndDeviceOp()
|
| D | execute.cc | 536 kernel.reset(new KernelAndDeviceOp( in GetOrCreateKernelAndDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 56 kernel->reset(new KernelAndDeviceOp(ctx.GetRendezvous(), ctx.LogMemory(), flr, in CreateUncachedKernelAndDeviceOp()
|