Searched refs:tensor_stack (Results 1 – 1 of 1) sorted by relevance
527 std::vector<int64> tensor_stack; in PrepareBackprop() local528 tensor_stack.reserve(target.size()); in PrepareBackprop()530 tensor_stack.push_back(t); in PrepareBackprop()533 while (!tensor_stack.empty()) { in PrepareBackprop()534 int64 tensor_id = tensor_stack.back(); in PrepareBackprop()535 tensor_stack.pop_back(); in PrepareBackprop()555 tensor_stack.push_back(it); in PrepareBackprop()