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

Function IsCpuCompatibleMatMul

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

Source from the content-addressed store, hash-verified

317}
318
319bool IsCpuCompatibleMatMul(const NodeDef* matmul) {
320 DCHECK(IsMatMul(*matmul)) << "Expected MatMul op";
321 return NodeIsOnCpu(matmul) && IsCpuCompatibleDataType(matmul);
322}
323
324bool IsCpuCompatibleDepthwiseConv2dNative(const NodeDef* dw_conv2d) {
325 DCHECK(IsDepthwiseConv2dNative(*dw_conv2d))

Callers 2

IsCpuCompatibleFunction · 0.85

Calls 3

IsMatMulFunction · 0.85
IsCpuCompatibleDataTypeFunction · 0.85
NodeIsOnCpuFunction · 0.70

Tested by

no test coverage detected