Searched refs:KernelAndDeviceFunc (Results 1 – 5 of 5) sorted by relevance
89 KernelAndDeviceFunc::~KernelAndDeviceFunc() { in ~KernelAndDeviceFunc()133 Status KernelAndDeviceFunc::InstantiateFunc(const bool log_device_placement, in InstantiateFunc()223 Status KernelAndDeviceFunc::Init(const bool log_device_placement, in Init()321 Status KernelAndDeviceFunc::Run( in Run()338 void KernelAndDeviceFunc::RunAsync( in RunAsync()408 tensorflow::Device* KernelAndDeviceFunc::OutputDevice(int idx) const { in OutputDevice()422 tensorflow::Device* KernelAndDeviceFunc::OutputResourceDevice(int idx) const { in OutputResourceDevice()433 Device* KernelAndDeviceFunc::InputDevice(int i) const { in InputDevice()
32 class TestKernelAndDeviceFunc final : public KernelAndDeviceFunc {36 : KernelAndDeviceFunc( in TestKernelAndDeviceFunc()
255 class KernelAndDeviceFunc : public KernelAndDevice {260 KernelAndDeviceFunc( in KernelAndDeviceFunc() function285 ~KernelAndDeviceFunc() override;
527 kernel.reset(new KernelAndDeviceFunc( in GetOrCreateKernelAndDevice()
950 core::RefCountPtr<KernelAndDeviceFunc> kernel = nullptr; in TEST_F()952 kernel.reset(new KernelAndDeviceFunc( in TEST_F()998 core::RefCountPtr<KernelAndDeviceFunc> kernel = nullptr; in TEST_F()1000 kernel.reset(new KernelAndDeviceFunc( in TEST_F()