Searched defs:SpmdPartitioningVisitor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| H A D | spmd_partitioner.h | 199 class SpmdPartitioningVisitor; variable |
| H A D | spmd_partitioner.cc | 1988 SpmdPartitioningVisitor::SpmdPartitioningVisitor( in SpmdPartitioningVisitor() function in xla::spmd::SpmdPartitioningVisitor |