Home
last modified time | relevance | path

Searched defs:replicate_op (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dhoist_replicate_invariant_resource_writes.cc59 tf_device::ReplicateOp replicate_op, in MoveTailWritesAfterReplicate()
104 tf_device::ReplicateOp replicate_op) { in GetTailWritesToReplicateInvariantResourceVars()
129 getOperation().walk([&](tf_device::ReplicateOp replicate_op) { in runOnOperation()
132 for (auto replicate_op : replicate_ops) { in runOnOperation() local
H A Dreplicate_to_island.cc142 tf_executor::IslandOp island_op, tf_device::ReplicateOp replicate_op, in ExpandReplicateIntoReplicas()
239 tf_device::ReplicateOp replicate_op) { in CreateIslandsFromReplicate()
322 auto replicate_op = in runOnOperation() local
H A Dreplicate_invariant_op_hoisting.cc46 void MakeShapeOpInvariant(tf_device::ReplicateOp replicate_op, int num_replicas, in MakeShapeOpInvariant()
124 void HoistReplicateInvariantOps(tf_device::ReplicateOp replicate_op) { in HoistReplicateInvariantOps()
H A Dtpu_reorder_replicate_and_partitioned_inputs.cc91 auto replicate_op = builder.create<TF::TPUReplicatedInputOp>( in ReorderReplicateAndPartitionedInputs() local
H A Dtpu_colocate_composite_resource_ops.cc95 tf_device::ReplicateOp replicate_op, OpBuilder* builder) { in ColocateCompositeResourceOpsInReplicate()
H A Dtpu_variable_runtime_reformatting.cc479 while_op.body().walk([&](tf_device::ReplicateOp replicate_op) { in runOnOperation()
H A Dtpu_cluster_formation.cc462 auto replicate_op = builder.create<tf_device::ReplicateOp>( in ReplicateCluster() local