Searched refs:ReshapeSharding (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util_test.cc | 41 ReshapeSharding(input_shape, output_shape, sharding); in TEST()
51 ReshapeSharding(input_shape, output_shape, sharding); in TEST()
62 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
74 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
88 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
101 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
114 ReshapeSharding(input_shape, output_shape, sharding); in TEST()
127 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
139 ReshapeSharding(input_shape, output_shape, input_sharding); in TEST()
147 absl::optional<HloSharding> result = ReshapeSharding(shape, shape, sharding); in TEST()
[all …]
|
| D | hlo_sharding_util.h | 89 absl::optional<HloSharding> ReshapeSharding(const Shape& source_shape,
|
| D | sharding_propagation.cc | 837 hlo_sharding_util::ReshapeSharding( in InferShardingFromOperands()
1248 return hlo_sharding_util::ReshapeSharding( in GetShardingFromUser()
|
| D | hlo_sharding_util.cc | 297 absl::optional<HloSharding> ReshapeSharding(const Shape& source_shape, in ReshapeSharding() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 831 auto padded_slice_sharding = hlo_sharding_util::ReshapeSharding( in PartitionBaseCase()
|
| D | spmd_partitioner.cc | 1978 auto desired_operand_sharding = hlo_sharding_util::ReshapeSharding( in HandleReshape()
|