Searched refs:assign_sharding (Results 1 – 2 of 2) sorted by relevance
109 xla::XlaScopedShardingAssignment assign_sharding(b, op_sharding); in Compute() local
210 xla::XlaScopedShardingAssignment assign_sharding( in BuildComputation() local283 xla::XlaScopedShardingAssignment assign_sharding( in BuildComputation() local821 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments() local830 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments() local