Searched refs:stb_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 60 bool ResolveDilatedConv(Model* model, Operator* conv_base_op, Operator* stb_op, in ResolveDilatedConv() argument
129 auto* pre_stb_pad_op = GetOpWithOutput(*model, stb_op->inputs[0]); in ResolveDilatedConv()
132 stb_op->inputs[0] = pre_stb_pad_op->inputs[0]; in ResolveDilatedConv()
151 post_stb_op->inputs[0] = stb_op->inputs[0]; in ResolveDilatedConv()
154 conv_op->inputs[0] = stb_op->inputs[0]; in ResolveDilatedConv()
161 DeleteOpAndArrays(model, stb_op); in ResolveDilatedConv()
174 auto* stb_op = it->get(); in Run() local
179 if (stb_op->type != OperatorType::kSpaceToBatchND) { in Run()
182 if (stb_op->inputs.size() != 3) { in Run()
185 CHECK_EQ(stb_op->outputs.size(), 1); in Run()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 163 TF::SpaceToBatchNDOp stb_op = llvm::cast<TF::SpaceToBatchNDOp>(prev_op); in matchAndRewrite() local
199 stb_op.block_shape(), bts_op.block_shape(), expand_axis, rewriter); in matchAndRewrite()
203 if (stb_op.input().getType().dyn_cast<RankedTensorType>() == nullptr) { in matchAndRewrite()
236 auto stb_paddings = stb_op.paddings(); in matchAndRewrite()
265 expand_op.setOperand(0, stb_op.input()); in matchAndRewrite()
267 auto input_shape = stb_op.input().getType().cast<ShapedType>().getShape(); in matchAndRewrite()
273 expand_shape, getElementTypeOrSelf(stb_op.input())); in matchAndRewrite()
283 conv_result_shape, getElementTypeOrSelf(stb_op.input())); in matchAndRewrite()
295 op.setOperand(0, stb_op.input()); in matchAndRewrite()
303 stb_op.getResult().dropAllUses(); in matchAndRewrite()
|