Searched refs:KernelAndDeviceFunc (Results 1 – 5 of 5) sorted by relevance
90 KernelAndDeviceFunc::~KernelAndDeviceFunc() { in ~KernelAndDeviceFunc()139 Status KernelAndDeviceFunc::InstantiateFunc(const bool log_device_placement, in InstantiateFunc()244 Status KernelAndDeviceFunc::Init(const bool log_device_placement, in Init()352 KernelAndDeviceFunc::PrepareForRun( in PrepareForRun()412 Status KernelAndDeviceFunc::Run( in Run()457 void KernelAndDeviceFunc::RunAsync( in RunAsync()488 tensorflow::Device* KernelAndDeviceFunc::OutputDevice(int idx) const { in OutputDevice()502 tensorflow::Device* KernelAndDeviceFunc::OutputResourceDevice(int idx) const { in OutputResourceDevice()513 Device* KernelAndDeviceFunc::InputDevice(int i) const { in InputDevice()
32 class TestKernelAndDeviceFunc final : public KernelAndDeviceFunc {36 : KernelAndDeviceFunc( in TestKernelAndDeviceFunc()
265 class KernelAndDeviceFunc : public KernelAndDevice {270 KernelAndDeviceFunc( in KernelAndDeviceFunc() function306 ~KernelAndDeviceFunc() override;
1223 kernel.reset(new KernelAndDeviceFunc( in GetOrCreateKernelAndDevice()
956 core::RefCountPtr<KernelAndDeviceFunc> kernel = nullptr; in TEST_F()958 kernel.reset(new KernelAndDeviceFunc( in TEST_F()1010 core::RefCountPtr<KernelAndDeviceFunc> kernel = nullptr; in TEST_F()1012 kernel.reset(new KernelAndDeviceFunc( in TEST_F()