Searched refs:assign_sharding (Results 1 – 4 of 4) sorted by relevance
63 xla::XlaScopedShardingAssignment assign_sharding(ctx->builder(), in Compile() local75 xla::XlaScopedShardingAssignment assign_sharding(ctx->builder(), manual); in Compile() local118 xla::XlaScopedShardingAssignment assign_sharding(ctx->builder(), manual); in Compile() local126 xla::XlaScopedShardingAssignment assign_sharding(ctx->builder(), in Compile() local
137 xla::XlaScopedShardingAssignment assign_sharding(b, op_sharding); in Compute() local
240 xla::XlaScopedShardingAssignment assign_sharding(builder, sharding); in BuildComputation() local339 xla::XlaScopedShardingAssignment assign_sharding(builder, sharding); in BuildComputation() local395 xla::XlaScopedShardingAssignment assign_sharding(builder, op_sharding); in BuildComputation() local1083 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments() local1096 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments() local
137 xla::XlaScopedShardingAssignment assign_sharding(b, device_sharding); in Compile() local