Searched refs:biasadd_op (Results 1 – 2 of 2) sorted by relevance
178 TF::BiasAddOp biasadd_op; in matchAndRewrite() local182 biasadd_op = llvm::cast<TF::BiasAddOp>(next_op); in matchAndRewrite()191 biasadd_op = llvm::cast<TF::BiasAddOp>(next_op); in matchAndRewrite()289 if (biasadd_op) { in matchAndRewrite()290 biasadd_op.setOperand(0, squeeze_op.output()); in matchAndRewrite()291 biasadd_op.output().setType(squeeze_op.output().getType()); in matchAndRewrite()294 if (biasadd_op) biasadd_op.setOperand(0, op.output()); in matchAndRewrite()
432 tensorflow::NodeDef* biasadd_op = tensorflow_graph->add_node(); in ConvertConvOperator() local433 biasadd_op->set_op("BiasAdd"); in ConvertConvOperator()434 biasadd_op->set_name(src_op.outputs[0]); in ConvertConvOperator()435 biasadd_op->add_input(conv_output); in ConvertConvOperator()436 biasadd_op->add_input(src_op.inputs[2]); in ConvertConvOperator()437 (*biasadd_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertConvOperator()523 tensorflow::NodeDef* biasadd_op = tensorflow_graph->add_node(); in ConvertDepthwiseConvOperator() local524 biasadd_op->set_op("BiasAdd"); in ConvertDepthwiseConvOperator()525 biasadd_op->set_name(src_op.outputs[0]); in ConvertDepthwiseConvOperator()526 biasadd_op->add_input(conv_output); in ConvertDepthwiseConvOperator()[all …]