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

Function IsKernelRegisteredForNode

tensorflow/core/grappler/utils.cc:549–562  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

547}
548
549Status IsKernelRegisteredForNode(
550 absl::string_view node_name, bool has_experimental_debug_info,
551 const NodeDef_ExperimentalDebugInfo& experimental_debug_info,
552 absl::string_view node_op, absl::string_view node_device,
553 AttrSlice node_attrs) {
554 DeviceNameUtils::ParsedName parsed_name;
555 if (!DeviceNameUtils::ParseFullName(node_device, &parsed_name)) {
556 return errors::InvalidArgument("Could not parse device name: ",
557 node_device);
558 }
559 return FindKernelDef(DeviceType(parsed_name.type), node_name,
560 has_experimental_debug_info, experimental_debug_info,
561 node_op, node_device, node_attrs, nullptr, nullptr);
562}
563
564Status IsKernelRegisteredForNode(const NodeDef& node) {
565 return IsKernelRegisteredForNode(node.name(),

Callers 5

TESTFunction · 0.85
OptimizeMethod · 0.85
TrySimplifyMethod · 0.85

Calls 8

InvalidArgumentFunction · 0.85
FindKernelDefFunction · 0.85
DeviceTypeClass · 0.85
AttrSliceClass · 0.85
attrMethod · 0.80
nameMethod · 0.65
opMethod · 0.45
deviceMethod · 0.45

Tested by 1

TESTFunction · 0.68