Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreplicate_to_island.cc140 tf_executor::IslandOp island_op, tf_device::ReplicateOp replicate_op, in ExpandReplicateIntoReplicas()
237 tf_device::ReplicateOp replicate_op) { in CreateIslandsFromReplicate()
320 auto replicate_op = in runOnFunction() local
Dreplicate_invariant_op_hoisting.cc76 void MakeShapeOpInvariant(tf_device::ReplicateOp replicate_op, int num_replicas, in MakeShapeOpInvariant()
154 void HoistReplicateInvariantOps(tf_device::ReplicateOp replicate_op) { in HoistReplicateInvariantOps()
Dtf_device_replication_pass.cc42 module.walk([&](tf_device::ReplicateOp replicate_op) { in runOnOperation()
Dtpu_reorder_replicate_and_partitioned_inputs.cc95 auto replicate_op = builder.create<TF::TPUReplicatedInputOp>( in ReorderReplicateAndPartitionedInputs() local
Dtpu_colocate_composite_resource_ops.cc92 tf_device::ReplicateOp replicate_op, OpBuilder* builder) { in ColocateCompositeResourceOpsInReplicate()
Dexecutor_tpuv1_outline_tpu_island.cc80 getOperation().walk([&](TF::TPUReplicateMetadataOp replicate_op) { in runOnOperation()
Dtpu_variable_runtime_reformatting.cc526 while_op.body().walk([&](tf_device::ReplicateOp replicate_op) { in runOnOperation()
Dtpu_cluster_formation.cc424 auto replicate_op = builder.create<tf_device::ReplicateOp>( in ReplicateCluster() local