Searched defs:xla_sharding_node (Results 1 – 1 of 1) sorted by relevance
196 void SetXlaShardingNodeAttr(Node* xla_sharding_node, int num_cores_per_replica, in SetXlaShardingNodeAttr()1075 Node* xla_sharding_node = nullptr; in InsertReshapeNodePairs() local