Searched refs:activations_to_use (Results 1 – 1 of 1) sorted by relevance
2415 HloInstruction* activations_to_use = nullptr; in PropagateOnBackpropFilterConv() local2417 activations_to_use = activations_new; in PropagateOnBackpropFilterConv()2419 activations_to_use = activations_chunks.back(); in PropagateOnBackpropFilterConv()2423 HaloDuplicateWithSlice(activations_to_use, spatial_dimension_to_split, in PropagateOnBackpropFilterConv()2446 HloInstruction* activations_to_use = nullptr; in PropagateOnBackpropFilterConv() local2449 activations_to_use = activations_new; in PropagateOnBackpropFilterConv()2453 activations_to_use, spatial_dimension_to_split, in PropagateOnBackpropFilterConv()2459 activations_slice = activations_to_use; in PropagateOnBackpropFilterConv()2462 activations_to_use = activations_chunks.back(); in PropagateOnBackpropFilterConv()2466 HaloDuplicateWithSlice(activations_to_use, spatial_dimension_to_split, in PropagateOnBackpropFilterConv()[all …]