Searched refs:activations_old (Results 1 – 1 of 1) sorted by relevance
1846 auto activations_old = convolution->mutable_operand(0); in PropagateOnConv() local1848 CHECK(old_to_new_instrs_.contains(activations_old)); in PropagateOnConv()1849 auto activations_new = old_to_new_instrs_[activations_old]; in PropagateOnConv()1878 activations_old->shape().dimensions(old_batch_dim); in PropagateOnConv()1900 SelectValidPortion(activations_new, activations_old, select_val, in PropagateOnConv()2148 auto activations_old = convolution->mutable_operand(0); in PropagateOnBackpropFilterConv() local2164 int64 old_split_dim_size = activations_old->shape().dimensions(old_space_dim); in PropagateOnBackpropFilterConv()2170 activations_old->shape().dimensions(old_batch_dim); in PropagateOnBackpropFilterConv()2173 old_to_new_instrs_.contains(activations_old)); in PropagateOnBackpropFilterConv()2181 if (!old_to_new_instrs_.contains(activations_old)) { in PropagateOnBackpropFilterConv()[all …]