Searched defs:replicate_block (Results 1 – 2 of 2) sorted by relevance
77 Block* replicate_block, TF::ShapeOp shape_op) { in MakeShapeOpInvariant()156 Block* replicate_block = &replicate_op.GetBody(); in HoistReplicateInvariantOps() local
61 Block* replicate_block = &replicate.GetBody(); in GetRemappedReplicatedInputIndices() local