Searched refs:activations_new (Results 1 – 1 of 1) sorted by relevance
613 TF_ASSIGN_OR_RETURN(HloInstruction * activations_new, in IncreaseSpatialSizeOnSpaceToBatchedShape()616 VLOG(3) << "Size increased activations " << activations_new->ToString(); in IncreaseSpatialSizeOnSpaceToBatchedShape()618 return activations_new; in IncreaseSpatialSizeOnSpaceToBatchedShape()1849 auto activations_new = old_to_new_instrs_[activations_old]; in PropagateOnConv() local1850 auto permute_dims = instr_to_dim_permute_map_[activations_new]; in PropagateOnConv()1890 activations_new, permuted_conv_dims_numbers, in PropagateOnConv()1892 activations_new = retval.instr; in PropagateOnConv()1896 LiteralUtil::Zero(activations_new->shape().element_type()))); in PropagateOnConv()1899 activations_new, in PropagateOnConv()1900 SelectValidPortion(activations_new, activations_old, select_val, in PropagateOnConv()[all …]