Home
last modified time | relevance | path

Searched defs:use_spmd (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_sharding_identification_pass.cc120 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationInputs()
247 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationOutputs()
315 bool use_spmd = false; in IdentifyXlaShardingForTPUComputation() local
Dtpu_resource_partitioning.cc75 bool use_spmd = false; in PartitionResourceReadsWrites() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1974 const bool use_spmd = (UseSpmdForXlaPartitioning(replicate_node) || in AssignArgsAndRetvalsToCores() local
2327 const bool use_spmd = in BuildCompileNode() local