Searched refs:sharding_builder (Results 1 – 7 of 7) sorted by relevance
179 name = "sharding_builder",180 srcs = ["sharding_builder.cc"],181 hdrs = ["sharding_builder.h"],214 ":sharding_builder",
29 namespace sharding_builder {
19 namespace sharding_builder { namespace
1315 OpSharding sharding = sharding_builder::AssignDevice(0); in Infeed()1334 sharding_builder::AssignDevice(0); in Infeed()
76 xla::sharding_builder::AssignDevice(core)); in ParseShardingFromDevice()
213 : xla::sharding_builder::AssignDevice(it->second)); in BuildComputation()285 : xla::sharding_builder::AssignDevice(core)); in BuildComputation()798 xla::sharding_builder::AssignDevice(core); in BuildArguments()823 : xla::sharding_builder::AssignDevice(core)); in BuildArguments()832 : xla::sharding_builder::AssignDevice(core)); in BuildArguments()
272 "//tensorflow/compiler/xla/client:sharding_builder",