Home
last modified time | relevance | path

Searched refs:LiftConditionsThroughReduction (Results 1 – 1 of 1) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/autodiff/
H A Dad_simplify.cc842 std::pair<PrimExpr, PrimExpr> LiftConditionsThroughReduction(const PrimExpr& cond, in LiftConditionsThroughReduction() function
1187 LiftConditionsThroughReduction(red->condition, red->axis, axis); in RemoveJacobianAndLiftNonzeroCondImpl()
1196 std::tie(outer_nz_cond, nz_cond) = LiftConditionsThroughReduction(nz_cond, red->axis, axis); in RemoveJacobianAndLiftNonzeroCondImpl()