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

Function Matmul

tensorflow/core/graph/testlib.cc:159–169  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

157}
158
159Node* Matmul(Graph* g, Node* in0, Node* in1, bool transpose_a,
160 bool transpose_b) {
161 Node* ret;
162 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "MatMul")
163 .Input(in0)
164 .Input(in1)
165 .Attr("transpose_a", transpose_a)
166 .Attr("transpose_b", transpose_b)
167 .Finalize(g, &ret));
168 return ret;
169}
170
171Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) {
172 Node* ret;

Callers 15

TEST_FFunction · 0.70
testNoDeviceGraphMethod · 0.50
testNestedDevicesMethod · 0.50
MatmulWrapMethod · 0.50
InitializeMethod · 0.50
TESTFunction · 0.50
TestHWAcceleratorFunction · 0.50
InitializeMethod · 0.50
TEST_FFunction · 0.50
CreateGraphDefFunction · 0.50
InitializeMethod · 0.50
TEST_FFunction · 0.50

Calls 5

NodeBuilderClass · 0.70
FinalizeMethod · 0.45
AttrMethod · 0.45
InputMethod · 0.45
NewNameMethod · 0.45

Tested by

no test coverage detected