Searched defs:dim_args (Results 1 – 3 of 3) sorted by relevance
290 const py::function& func) { in initTensorExprBindings()349 const std::vector<ExprHandle>& dim_args, in initTensorExprBindings()360 const std::vector<ExprHandle>& dim_args, in initTensorExprBindings()370 const std::vector<ExprHandle>& dim_args, in initTensorExprBindings()381 const std::vector<ExprHandle>& dim_args, in initTensorExprBindings()387 const std::vector<ExprHandle>& reduce_args) { in initTensorExprBindings()
36 std::vector<int32_t> dim_args = {0, 0, 0, 0}; in flip() local
6890 std::vector<ExprHandle> dim_args; in TEST() local