Home
last modified time | relevance | path

Searched refs:bidirectional_windowed_einsum (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc633 (!(options.bidirectional_windowed_einsum && num_partitions % 4 == 0) || in PartitionBaseCase()
639 if (options.bidirectional_windowed_einsum && num_partitions % 4 == 0 && in PartitionBaseCase()
1145 if (options.bidirectional_windowed_einsum && num_partitions % 4 == 0) { in PartitionBaseCase()
1411 (options.bidirectional_windowed_einsum && num_partitions % 4 == 0) in PartitionBaseCase()
1429 if (((options.bidirectional_windowed_einsum && num_partitions % 4 == 0) || in PartitionBaseCase()
1440 if (options.bidirectional_windowed_einsum && num_partitions % 4 == 0) { in PartitionBaseCase()
3414 if (!options.bidirectional_windowed_einsum && in DoCodeMotionForWindowedDotGeneralLoops()
Dspmd_partitioner.h53 bool bidirectional_windowed_einsum = false; member