Searched refs:default_sharding (Results 1 – 3 of 3) sorted by relevance
224 HloSharding default_sharding = HloSharding::Replicate(); in TEST_F() local227 default_sharding); in TEST_F()235 default_sharding); in TEST_F()244 default_sharding); in TEST_F()248 default_sharding); in TEST_F()258 default_sharding); in TEST_F()262 default_sharding); in TEST_F()
276 HloSharding default_sharding = in HandleBatchNormTraining() local284 inst->set_sharding(default_sharding); in HandleBatchNormTraining()363 HloSharding default_sharding = in HandleBatchNormInference() local371 inst->set_sharding(default_sharding); in HandleBatchNormInference()540 HloSharding default_sharding = in HandleBatchNormGrad() local548 inst->set_sharding(default_sharding); in HandleBatchNormGrad()
501 NodeDef default_sharding; in AddReshardOp() local502 default_sharding.set_op("Const"); in AddReshardOp()503 default_sharding.set_name(graph->NewName(strings::StrCat( in AddReshardOp()505 AddNodeAttr("dtype", DT_STRING, &default_sharding); in AddReshardOp()512 (*default_sharding.mutable_attr())["value"].mutable_tensor()); in AddReshardOp()514 Node* default_sharding_node = graph->AddNode(default_sharding, &status); in AddReshardOp()