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