Searched refs:GetShardingFromNodeDef (Results 1 – 4 of 4) sorted by relevance
92 GetShardingFromNodeDef(node_def, add_metadata)); in ParseShardingFromDevice()107 GetShardingFromNodeDef(node.def(), add_metadata)); in ParseShardingFromDevice()150 xla::StatusOr<absl::optional<xla::OpSharding>> GetShardingFromNodeDef( in GetShardingFromNodeDef() function
53 xla::StatusOr<absl::optional<xla::OpSharding>> GetShardingFromNodeDef(
105 return GetShardingFromNodeDef(node_def, /*add_metadata=*/true); in TEST_P()
2004 GetShardingFromNodeDef(input_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()2022 GetShardingFromNodeDef(input_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()2136 GetShardingFromNodeDef(output_node->def(), /*add_metadata=*/true)); in AssignArgsAndRetvalsToCores()