Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_sharding_identification_pass.cc182 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationInputs()
352 StringRef logical_core_0_sharding, bool use_spmd, in IdentifyXlaShardingForComputationOutputs()
424 bool use_spmd = false; in IdentifyXlaShardingForTPUComputation() local
H A Dtpu_resource_partitioning.cc76 bool use_spmd = false; in PartitionResourceReadsWrites() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc2187 const bool use_spmd = (UseSpmdForXlaPartitioning(replicate_node) || in AssignArgsAndRetvalsToCores() local
2621 const bool use_spmd = in BuildCompileNode() local
3270 const bool use_spmd = in BuildExecuteNodes() local