Searched defs:bias_add_op (Results 1 – 7 of 7) sorted by relevance
55 mlir::TF::BiasAddOp bias_add_op = llvm::cast<mlir::TF::BiasAddOp>(op); in ExpandOp() local123 mlir::TF::BiasAddOp bias_add_op = llvm::cast<mlir::TF::BiasAddOp>(op); in ComputeLayoutForward() local160 mlir::TF::BiasAddOp bias_add_op = llvm::cast<mlir::TF::BiasAddOp>(op); in ComputeLayoutBackward() local
141 Operation *contraction_op, Operation *bias_add_op) { in GetContractionBiasAddOpState()250 Operation *bias_add_op = op->getOperand(0).getDefiningOp(); in matchPattern() local279 Operation *&bias_add_op = pattern.bias_add; in matchAndRewrite() local
117 Operator* bias_add_op = !has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv() local
199 auto *bias_add_op = bias_add.getOperation(); in matchAndRewrite() local
102 auto* bias_add_op = graph_builder->AddNode(matmul_op->GetTFLiteNodeID()); in AddFullyConnectedHelper() local
358 Output bias_add_op = in TestQuantizeBiasAdd() local889 Output bias_add_op = in TestConvertFakeQuantsToRequantize() local1041 Output bias_add_op = in TestConvertFakeQuantsEndToEnd() local1090 Output bias_add_op = in TestHoistFakeQuants() local
3805 auto bias_add_op = cudnn_frontend::OperationBuilder( in GetCudnnFusedOperationGraph() local