root 83 src/AssociativeOpsTable.cpp RootExpr root; root 85 src/AssociativeOpsTable.cpp TableKey(ValType t, RootExpr r, size_t d) : type(t), root(r), dim(d) {} root 88 src/AssociativeOpsTable.cpp return (type == other.type) && (root == other.root) && (dim == other.dim); root 96 src/AssociativeOpsTable.cpp if (root < other.root) { root 98 src/AssociativeOpsTable.cpp } else if (root > other.root) { root 294 src/AssociativeOpsTable.cpp const vector<AssociativePattern> &get_ops_table_helper(Type t, RootExpr root, size_t dim) { root 295 src/AssociativeOpsTable.cpp TableKey gen_key(ValType::All, root, dim); root 296 src/AssociativeOpsTable.cpp TableKey key(convert_halide_type_to_val_type(t), root, dim); root 336 src/AssociativeOpsTable.cpp RootExpr root = RootExpr::Unknown; root 339 src/AssociativeOpsTable.cpp root = RootExpr::Add; root 342 src/AssociativeOpsTable.cpp root = RootExpr::Sub; root 345 src/AssociativeOpsTable.cpp root = RootExpr::Mul; root 348 src/AssociativeOpsTable.cpp root = RootExpr::Min; root 351 src/AssociativeOpsTable.cpp root = RootExpr::Max; root 354 src/AssociativeOpsTable.cpp root = RootExpr::Select; root 357 src/AssociativeOpsTable.cpp root = RootExpr::And; root 360 src/AssociativeOpsTable.cpp root = RootExpr::Or; root 363 src/AssociativeOpsTable.cpp root = RootExpr::Cast; root 366 src/AssociativeOpsTable.cpp if (root != RootExpr::Unknown) { root 367 src/AssociativeOpsTable.cpp const vector<AssociativePattern> &table = get_ops_table_helper(exprs[0].type(), root, exprs.size()); root 2318 src/Func.cpp return compute_at(LoopLevel::root()); root 2336 src/Func.cpp return store_at(LoopLevel::root()); root 775 src/Generator.cpp {"root", LoopLevel::root()}, root 140 src/Schedule.h EXPORT static LoopLevel root(); root 808 src/ScheduleFunctions.cpp loop_level = LoopLevel::root(); root 923 src/ScheduleFunctions.cpp ends_with(op->name, LoopLevel::root().to_string())) { root 1151 src/ScheduleFunctions.cpp string root_var = LoopLevel::root().to_string(); root 155 src/ScheduleParam.h set(LoopLevel::root()); root 61 test/correctness/schedule_param.cpp compute_at.set(LoopLevel::root()); // this value will not be used root 79 test/generator/nested_externs_generator.cpp Output<Buffer<float>> root{ "root", 3 }; root 86 test/generator/nested_externs_generator.cpp root(x, y, c) = extern_stage_combine(x, y, c); root 92 test/generator/nested_externs_generator.cpp f.compute_at(root, y).reorder_storage(args[2], args[0], args[1]); root 94 test/generator/nested_externs_generator.cpp set_interleaved(root); root 95 test/generator/nested_externs_generator.cpp root.reorder_storage(c, x, y); root 30 test/generator/stubtest_generator.cpp ScheduleParam<LoopLevel> intermediate_level{ "intermediate_level", LoopLevel::root() };