Searched refs:KernelAndDeviceOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 57 KernelAndDeviceOp::~KernelAndDeviceOp() { in ~KernelAndDeviceOp()
67 Status KernelAndDeviceOp::Init(const NodeDef& ndef, in Init()
151 Status KernelAndDeviceOp::Run(const gtl::InlinedVector<TensorValue, 4>& inputs, in Run()
206 Status KernelAndDeviceOp::Run(ScopedStepContainer* step_container, in Run()
359 tensorflow::Device* KernelAndDeviceOp::OutputDevice(int idx) const { in OutputDevice()
373 tensorflow::Device* KernelAndDeviceOp::OutputResourceDevice(int idx) const { in OutputResourceDevice()
387 DataType KernelAndDeviceOp::input_type(int i) const { in input_type()
395 Device* KernelAndDeviceOp::InputDevice(int i) const { in InputDevice()
|
| D | kernel_and_device.h | 130 class KernelAndDeviceOp final : public KernelAndDevice {
132 KernelAndDeviceOp( in KernelAndDeviceOp() function
143 virtual ~KernelAndDeviceOp();
|
| D | kernel_and_device_test.cc | 119 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceInit()
142 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceRun()
|
| D | execute.cc | 451 kernel = new KernelAndDeviceOp( in EagerLocalExecute()
|