Searched refs:ParseShardingFromDevice (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util_test.cc | 22 TEST(CoreUtilTest, ParseShardingFromDevice) { in TEST() argument
36 auto parse_status = ParseShardingFromDevice("", 1); in TEST()
39 parse_status = ParseShardingFromDevice("", 100); in TEST()
43 parse_status = ParseShardingFromDevice("/device:A_REPLICATED_CORE:-1", 100); in TEST()
46 parse_status = ParseShardingFromDevice("/device:A_REPLICATED_CORE:55", 100); in TEST()
50 parse_status = ParseShardingFromDevice("/device:A_REPLICATED_CORE:100", 100); in TEST()
53 parse_status = ParseShardingFromDevice("/cpu:0", 100); in TEST()
|
| D | sharding_util.cc | 52 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
80 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
85 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding); in ParseShardingFromDevice()
88 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice( in ParseShardingFromDevice() function
96 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding); in ParseShardingFromDevice()
|
| D | sharding_util.h | 36 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice(
40 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice(
43 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice(
|
| D | tf2xla_util_test.cc | 243 auto parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
251 parse_status = ParseShardingFromDevice(*c_node, num_cores_per_replica); in TEST()
258 parse_status = ParseShardingFromDevice(*a_node, num_cores_per_replica); in TEST()
|
| D | xla_compilation_device.cc | 100 auto sharding_parse_result = ParseShardingFromDevice( in Compute()
|
| D | tf2xla_util.cc | 508 ParseShardingFromDevice( in SetNodeShardingFromNeighbors()
|
| D | xla_compiler.cc | 83 ParseShardingFromDevice(*n, std::numeric_limits<int32>::max())); in ComputeArgAndRetvalCores()
|