Lines Matching refs:pad_before
175 const tvm::Array<tvm::Expr>& pad_before,
181 if (pad_after.size() < pad_before.size()) {
182 for (size_t i = pad_after.size(); i < pad_before.size(); ++i) {
183 pad_after.push_back(pad_before[i]);
186 CHECK_GE(pad_before.size(), 1);
187 CHECK_EQ(pad_before.size(), pad_after.size());
191 for (const auto &ele : pad_before) {
198 if (i >= pad_before.size()) {
229 ovars[i] < pad_before[i],
231 tvm::if_then_else(ovars[i] >= pad_before[i] + t->shape[i],
233 ovars[i] - pad_before[i])));
236 ovars[i] < pad_before[i],
237 pad_before[i] - ovars[i],
238 tvm::if_then_else(ovars[i] >= pad_before[i] + t->shape[i],
239 t->shape[i] * 2 - ovars[i] + pad_before[i] - 2,
240 ovars[i] - pad_before[i])));