Function IsGpuCompatibleMatMul

tensorflow/core/grappler/optimizers/remapper.cc:312–317

}

bool IsGpuCompatibleMatMul(const RemapperContext& ctx,
    const NodeDef* matmul) {
  DCHECK(IsMatMul(*matmul)) << "Expected MatMul op";
  return BlasLtMatmulEnabled() && NodeIsOnGpu(matmul) &&
         IsGpuCompatibleDataType(matmul);
}

bool IsCpuCompatibleMatMul(const NodeDef* matmul) {
  DCHECK(IsMatMul(*matmul)) << "Expected MatMul op";