Searched refs:no_sharding (Results 1 – 2 of 2) sorted by relevance
37 xla::XlaScopedShardingAssignment no_sharding(ctx->builder(), in Compile() local
342 py::class_<NoSharding> no_sharding(pmap_lib, "NoSharding"); in BuildPmapSubmodule() local343 no_sharding.def(py::init<>()) in BuildPmapSubmodule()