Searched defs:arg_dims (Results 1 – 2 of 2) sorted by relevance
315 auto arg_dims = llvm::drop_begin(arg_ty.getShape(), 1); in fold() local
5419 std::vector<int64_t> arg_dims = *arg->mutable_dimensions(); in HandleReduce() local