Searched refs:partitioned_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2307 auto partitioned_shape = in HandleDynamicSlice() local
2310 partitioned_shape, new_input, new_indices, in HandleDynamicSlice()
2311 partitioned_shape.dimensions())); in HandleDynamicSlice()
2375 const auto& partitioned_shape = partitioned_input->shape(); in HandleDynamicUpdateSlice() local
2384 const int64 per_partition_size = partitioned_shape.dimensions(dim); in HandleDynamicUpdateSlice()
2441 partitioned_shape, partitioned_input, replicate_update, new_indices)); in HandleDynamicUpdateSlice()
2467 auto partitioned_shape = in HandleDynamicUpdateSlice() local
2470 partitioned_shape, new_input, new_update, new_indices)); in HandleDynamicUpdateSlice()
2524 auto partitioned_shape = in HandleInfeed() local
2530 partitioned_shape, in HandleInfeed()
[all …]
|