Searched refs:KernelAndDevice (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 53 class KernelAndDevice {
65 KernelAndDevice( in KernelAndDevice() function
78 virtual ~KernelAndDevice() {} in ~KernelAndDevice()
130 class KernelAndDeviceOp final : public KernelAndDevice {
138 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
188 class KernelAndDeviceFunc final : public KernelAndDevice {
199 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
|
| D | execute_node.h | 36 KernelAndDevice* kernel, NodeExecStats* maybe_stats, in ExecuteNode()
82 tensorflow::KernelAndDevice* kernel_;
|
| D | context.h | 153 KernelAndDevice* GetCachedKernel(Fprint128 cache_key);
155 void AddKernelToCache(Fprint128 cache_key, KernelAndDevice* kernel);
284 std::unordered_map<Fprint128, KernelAndDevice*, Fprint128Hasher> kernel_cache_
|
| D | execute.h | 48 KernelAndDevice* kernel, NodeExecStats* maybe_stats,
|
| D | context.cc | 350 KernelAndDevice* EagerContext::GetCachedKernel(Fprint128 cache_key) { in GetCachedKernel()
356 KernelAndDevice* kernel) { in AddKernelToCache()
|
| D | execute.cc | 183 const KernelAndDevice* kernel, in ValidateInputTypeAndPlacement()
389 KernelAndDevice* kernel = ctx->GetCachedKernel(cache_key); in EagerLocalExecute()
914 KernelAndDevice* kernel, NodeExecStats* maybe_stats, in EagerKernelExecute()
|