Home
last modified time | relevance | path

Searched defs:ParseShardingFromDevice (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dsharding_util.cc57 StatusOr<std::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
89 StatusOr<std::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
101 StatusOr<std::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
H A Dsharding_util_test.cc24 TEST(CoreUtilTest, ParseShardingFromDevice) { in TEST() argument