Searched refs:SpmdPartitionerOptions (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 36 struct SpmdPartitionerOptions { struct
185 SpmdPartitionerOptions options);
187 SpmdPartitionerOptions options, in SpmdPartitioner()
223 const SpmdPartitionerOptions& options() { return options_; } in options()
230 SpmdPartitionerOptions options);
249 SpmdPartitionerOptions options_;
406 SpmdPartitionerOptions options, SpmdPartitioner* partitioner);
499 const SpmdPartitionerOptions& options);
520 HloComputation* computation, const SpmdPartitionerOptions& options);
545 const SpmdPartitionerOptions options_;
|
| D | convolution_handler.h | 36 int64 num_partitions, const SpmdPartitionerOptions& options,
|
| D | dot_handler.cc | 480 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionBaseCase()
1644 const SpmdPartitionerOptions& options, SpmdBuilder* b,
1659 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionDotGroupOnBatch()
1993 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionDotGroupOnNonContracting()
2088 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionDotGroupOnContracting()
2290 const HloSharding& output_sharding, const SpmdPartitionerOptions& options, in LhsIsBestMatchForNonContractingPartitioning()
2440 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionDot()
2806 const SpmdPartitionerOptions& options, SpmdBuilder* b, in PartitionDot()
3401 HloComputation* computation, const SpmdPartitionerOptions& options) { in DoCodeMotionForWindowedDotGeneralLoops()
|
| D | convolution_handler.cc | 823 int64 num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolutionBaseCase()
971 int64 num_partitions, const SpmdPartitionerOptions& options, in PartitionConvolution()
|
| D | spmd_partitioner.cc | 1354 int64* next_channel_id, SpmdLogger* logger, SpmdPartitionerOptions options, in SpmdPartitioningVisitor()
3383 const SpmdPartitionerOptions& options) { in DoPartition()
3494 SpmdPartitionerOptions options) in SpmdPartitioner()
3645 SpmdPartitionerOptions options) { in CreateVisitor()
|
| D | spmd_partitioner_test.cc | 44 SpmdPartitionerOptions options; in PartitionComputation()
|