Searched defs:matmul_op (Results 1 – 11 of 11) sorted by relevance
50 OpBuilder* matmul_op, in AddFullyConnectedHelper()255 auto* matmul_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local269 OpBuilder* matmul_op) { in AddFullyConnected()
191 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local224 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
67 const auto* matmul_op = in Run() local
80 auto matmul_op = llvm::dyn_cast<TF::BatchMatMulV2Op>(op); in RewriteBatchMatMulV2Op() local
165 TFE_Op* matmul_op; in TEST_F() local
215 Output matmul_op = in TestFoldBatchNormsMatMul() local
504 auto matmul_op = AddMatMul(scope, "matmul", input); in TEST_P() local
457 auto* matmul_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
649 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local1530 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
1286 auto matmul_op = output_view.GetNode("allow1"); in TEST_F() local
1155 auto matmul_op = s.WithOpName("matmul"); in TEST_F() local