Searched refs:padding_attr (Results 1 – 2 of 2) sorted by relevance
523 auto padding_attr = primitive->GetAttr("paddings"); in InferImplPad() local524 MS_EXCEPTION_IF_NULL(padding_attr); in InferImplPad()525 if (!padding_attr->isa<ValueTuple>()) { in InferImplPad()528 std::vector<ValuePtr> paddings = padding_attr->cast<ValueTuplePtr>()->value(); in InferImplPad()
253 std::vector<std::vector<int64_t>> padding_attr; in ChangeOpPad() local254 (void)std::transform(padding_list.begin(), padding_list.end(), std::back_inserter(padding_attr), in ChangeOpPad()256 prim->AddAttr(ops::kPaddings, MakeValue(padding_attr)); in ChangeOpPad()