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

Function GetKernel

tensorflow/core/common_runtime/ring_gatherer_test.cc:100–111  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

98};
99
100std::unique_ptr<OpKernel> GetKernel(const NodeDef& node,
101 const DeviceType& device_type,
102 DeviceBase* device) {
103 Status status;
104 std::unique_ptr<OpKernel> k = CreateOpKernel(
105 device_type, device, device->GetAllocator(AllocatorAttributes()), node,
106 TF_GRAPH_DEF_VERSION, &status);
107 if (!status.ok()) {
108 LOG(FATAL) << status;
109 }
110 return k;
111}
112
113static int64 kStepId = 123;
114

Callers 1

GetCollectiveGatherMethod · 0.70

Calls 4

CreateOpKernelFunction · 0.85
AllocatorAttributesClass · 0.85
GetAllocatorMethod · 0.45
okMethod · 0.45

Tested by

no test coverage detected