Searched refs:activations_slice (Results 1 – 1 of 1) sorted by relevance
2422 HloInstruction * activations_slice, in PropagateOnBackpropFilterConv()2428 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()2447 HloInstruction* activations_slice = nullptr; in PropagateOnBackpropFilterConv() local2451 TF_ASSIGN_OR_RETURN(activations_slice, in PropagateOnBackpropFilterConv()2459 activations_slice = activations_to_use; in PropagateOnBackpropFilterConv()2465 activations_slice, in PropagateOnBackpropFilterConv()2473 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()2482 HloInstruction * activations_slice, in PropagateOnBackpropFilterConv()2487 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()