Searched refs:matmul2 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/profiler/ |
| D | profiler_test.cc | 164 const MultiGraphNodeProto* matmul2 = ExtractNode(ret3, "MatMul"); in TEST_F() local
165 EXPECT_TRUE(matmul2); in TEST_F()
166 EXPECT_GT(matmul2->exec_micros(), 0); in TEST_F()
168 EXPECT_GT(matmul2->accelerator_exec_micros(), 0); in TEST_F()
170 EXPECT_EQ(matmul2->accelerator_exec_micros(), 0); in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 923 TFE_Op* matmul2 = MatMulOp(ctx, m1, m1); in Execute_MatMul_CPU_Runtime_Error() local
924 TFE_OpSetDevice(matmul2, "/job:localhost/replica:0/task:0/device:CPU:0", in Execute_MatMul_CPU_Runtime_Error()
945 TFE_Execute(matmul2, &retvals[0], &num_retvals, status); in Execute_MatMul_CPU_Runtime_Error()
957 TFE_Execute(matmul2, &retvals[0], &num_retvals, status); in Execute_MatMul_CPU_Runtime_Error()
962 TFE_DeleteOp(matmul2); in Execute_MatMul_CPU_Runtime_Error()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1704 TF_Operation* matmul2 = MatMul(expected_graph_, s_, const0, const3, in BuildExpectedGraph() local
1707 expected_grad_outputs[1] = {matmul2, 0}; in BuildExpectedGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 521 Output matmul2 = ops::MatMul(s.WithOpName("matmul2"), zeros, y); in TEST_F() local
533 matmul2, add1, add2, bias_add1, bias_add2, sub1, sub2}); in TEST_F()
|