Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h50 bool unroll_windowed_einsum = false; member
Ddot_handler.cc1231 } else if (options.unroll_windowed_einsum && num_partitions % 2 == 0) { in PartitionBaseCase()
1430 (options.unroll_windowed_einsum && num_partitions % 2 == 0)) && in PartitionBaseCase()
1447 if (options.unroll_windowed_einsum && num_partitions % 2 == 0) { in PartitionBaseCase()
3415 !options.unroll_windowed_einsum && !loop.windowed_in_contracting_dims && in DoCodeMotionForWindowedDotGeneralLoops()