Searched defs:thunks (Results 1 – 13 of 13) sorted by relevance
50 private final Map<String, Supplier<ImportScope>> thunks; field in ImportIndex52 public ImportIndex(TurbineLogWithSource log, ImmutableMap<String, Supplier<ImportScope>> thunks) { in ImportIndex()62 Map<String, Supplier<ImportScope>> thunks = new HashMap<>(); in create() local
40 const std::vector<std::unique_ptr<Thunk>>& thunks() const { return thunks_; } in thunks() function
51 std::unique_ptr<ThunkSequence> thunks, in ThunkSchedule()84 ThunkSchedule::ThunkSchedule(std::unique_ptr<ThunkSequence> thunks) in ThunkSchedule()
42 std::vector<std::unique_ptr<Thunk>> thunks; in BuildGemmThunk() local
28 std::vector<std::unique_ptr<Thunk>> thunks) in SequentialThunk()
617 std::unique_ptr<ThunkSequence> thunks = in CompileModuleToLlvmIrImpl() local1127 std::unique_ptr<ThunkSequence> thunks = ir_emitter->ConsumeThunkSequence(); in CompileLmhloToExecutable() local
1277 std::vector<std::unique_ptr<Thunk>> thunks; in EmitGemmThunkFromMlir() local1506 std::vector<std::unique_ptr<Thunk>> thunks; in EmitCholeskyThunkFromMlir() local1624 std::vector<std::unique_ptr<Thunk>> thunks; in EmitTriangularSolveFromMlir() local1886 std::vector<std::unique_ptr<Thunk>> thunks; in HandleFusion() local2206 std::vector<std::unique_ptr<Thunk>> thunks; in EmitSelectAndScatterFromMlir() local2500 std::vector<std::unique_ptr<Thunk>> thunks; in EmitScatterFromMlir() local2860 std::vector<std::unique_ptr<Thunk>> thunks; in EmitSortFromMlir() local3154 std::vector<std::unique_ptr<Thunk>> thunks; in EmitNcclThunkFromMlir() local5544 std::vector<std::unique_ptr<Thunk>> thunks; in EmitReductionFromOrToContiguousDimensions() local
30 EXPORT_FROM_MOJO_CORE void MojoGetSystemThunks(MojoSystemThunks* thunks) { in MojoGetSystemThunks()
62 std::vector<Chunk *> thunks; variable
56 Object evaluate(EvaluationContext context, List<Node> thunks) { in evaluate()
195 private final ImmutableList<Node> thunks; field in DirectiveNode.MacroCallNode
179 llvm::DenseMap<Symbol *, Thunk *> thunks; variable
473 void MojoEmbedderSetSystemThunks(const MojoSystemThunks* thunks) { in MojoEmbedderSetSystemThunks()