Searched refs:activations_chunks (Results 1 – 1 of 1) sorted by relevance
2411 std::vector<HloInstruction*> activations_chunks; in PropagateOnBackpropFilterConv() local2419 activations_to_use = activations_chunks.back(); in PropagateOnBackpropFilterConv()2428 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()2432 absl::c_reverse(activations_chunks); in PropagateOnBackpropFilterConv()2462 activations_to_use = activations_chunks.back(); in PropagateOnBackpropFilterConv()2473 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()2479 activations_to_use = activations_chunks.back(); in PropagateOnBackpropFilterConv()2487 activations_chunks.push_back(activations_slice); in PropagateOnBackpropFilterConv()2490 for (int64 i = 0; i < activations_chunks.size(); ++i) { in PropagateOnBackpropFilterConv()2492 activations_chunks[i]->shape().dimensions().begin(), in PropagateOnBackpropFilterConv()[all …]