Searched defs:static_argnums (Results 1 – 2 of 2) sorted by relevance
135 py::function get_jax_enable_x64, std::vector<int> static_argnums) in PmapFunction()426 std::vector<int> static_argnums) -> std::unique_ptr<PmapFunction> { in BuildPmapSubmodule()
185 absl::Span<int const> static_argnums, in ParseArguments()920 std::vector<int> static_argnums) in CompiledFunction()1224 std::vector<int> static_argnums) -> std::unique_ptr<CompiledFunction> { in BuildJaxjitSubmodule()