Searched defs:arg_dims (Results 1 – 2 of 2) sorted by relevance
297 auto arg_dims = llvm::drop_begin(arg_ty.getShape(), 1); in fold() local
4619 std::vector<int64> arg_dims = arg->dimensions(); in HandleReduce() local