Searched refs:bts_op (Results 1 – 2 of 2) sorted by relevance
177 TF::BatchToSpaceNDOp bts_op; in matchAndRewrite() local185 bts_op = llvm::cast<TF::BatchToSpaceNDOp>(next_op); in matchAndRewrite()188 bts_op = llvm::cast<TF::BatchToSpaceNDOp>(next_op); in matchAndRewrite()199 stb_op.block_shape(), bts_op.block_shape(), expand_axis, rewriter); in matchAndRewrite()237 auto bts_crops = bts_op.crops(); in matchAndRewrite()278 bts_op.output().getType().cast<ShapedType>().getShape(); in matchAndRewrite()286 squeeze_op.getResult().setType(bts_op.output().getType()); in matchAndRewrite()296 op.getResult().setType(bts_op.getResult().getType()); in matchAndRewrite()300 bts_op.getResult().replaceAllUsesWith(bts_op.input()); in matchAndRewrite()
109 const auto* bts_op = has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv() local110 if (bts_op->type != OperatorType::kBatchToSpaceND) { in ResolveDilatedConv()113 CHECK_EQ(bts_op->inputs.size(), 3); in ResolveDilatedConv()114 CHECK_EQ(bts_op->outputs.size(), 1); in ResolveDilatedConv()160 DeleteOpAndArrays(model, bts_op); in ResolveDilatedConv()