Searched defs:matmul_op (Results 1 – 10 of 10) 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()
189 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local221 auto* matmul_op = new TensorFlowMatMulOperator; in Run() local
66 const auto* matmul_op = in Run() local
78 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
93 AbstractOperationPtr matmul_op(ctx->CreateOperation()); in MatMul() local
457 auto* matmul_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
644 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local1525 tensorflow::NodeDef* matmul_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
727 auto matmul_op = s.WithOpName("matmul"); in TEST_F() local