Searched refs:GetShardingFromNodeDef (Results 1 – 5 of 5) sorted by relevance
93 GetShardingFromNodeDef(node_def, add_metadata)); in ParseShardingFromDevice()108 GetShardingFromNodeDef(node.def(), add_metadata)); in ParseShardingFromDevice()175 xla::StatusOr<std::optional<xla::OpSharding>> GetShardingFromNodeDef( in GetShardingFromNodeDef() function
53 StatusOr<std::optional<xla::OpSharding>> GetShardingFromNodeDef(
104 return GetShardingFromNodeDef(node_def, /*add_metadata=*/true); in TEST_P()
2218 GetShardingFromNodeDef(input_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()2236 GetShardingFromNodeDef(input_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()2360 GetShardingFromNodeDef(output_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()
1802 GetShardingFromNodeDef(variable->def(), /*add_metadata=*/false)); in ReplaceAndPartitionXLAShardingVariable()