MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsDeviceCompatible

Function IsDeviceCompatible

tensorflow/core/grappler/optimizers/remapper.cc:413–415  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

411// Returns true if the given pattern is supported on the assigned device.
412template <typename Pattern>
413bool IsDeviceCompatible(const RemapperContext& ctx, Pattern& matched) {
414 return IsCpuCompatible(ctx, matched) || IsGpuCompatible(ctx, matched);
415}
416
417bool IsSupportedActivation(const NodeDef& node) {
418 if (!DisableMKL()) {

Callers 5

FindContractionWithBiasFunction · 0.85
AddFusedContractionNodeFunction · 0.85
AddFusedConv2DNodeFunction · 0.85

Calls 2

IsCpuCompatibleFunction · 0.85
IsGpuCompatibleFunction · 0.85

Tested by

no test coverage detected