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

Function GetDiv

tensorflow/core/common_runtime/ring_reducer_test.cc:124–133  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

122}
123
124std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type,
125 DeviceBase* device) {
126 NodeDef node_def;
127 NodeDefBuilder builder("add_node", "Div");
128 TF_CHECK_OK(builder.Attr("T", dtype)
129 .Input(FakeInput(dtype))
130 .Input(FakeInput(dtype))
131 .Finalize(&node_def));
132 return GetKernel(node_def, device_type, device);
133}
134
135static int64 kStepId = 123;
136

Callers 1

DoReduceMethod · 0.70

Calls 5

GetKernelFunction · 0.70
FakeInputFunction · 0.50
FinalizeMethod · 0.45
InputMethod · 0.45
AttrMethod · 0.45

Tested by

no test coverage detected