Searched refs:conv2d_input (Results 1 – 1 of 1) sorted by relevance
594 auto conv2d_input = conv2d.input(); in HandleFirstConvolution() local595 if (auto block_arg = conv2d_input.dyn_cast<mlir::BlockArgument>()) { in HandleFirstConvolution()600 if (auto pad_op = dyn_cast_or_null<TF::PadOp>(conv2d_input.getDefiningOp())) { in HandleFirstConvolution()622 if (!conv2d_input.getDefiningOp()) return; in HandleFirstConvolution()623 for (Operation* user : conv2d_input.getDefiningOp()->getUsers()) { in HandleFirstConvolution()